From 46b06b13e7f92a05ffcfc64a87a3361467ae7cc6 Mon Sep 17 00:00:00 2001 From: Malcolm Keyes Date: Wed, 9 Aug 2023 18:21:16 -0400 Subject: [PATCH] fix: Access saved dataset proto correctly in list_saved_datasets Signed-off-by: Malcolm Keyes --- sdk/python/feast/infra/registry/proto_registry_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdk/python/feast/infra/registry/proto_registry_utils.py b/sdk/python/feast/infra/registry/proto_registry_utils.py index 2a275703dbb..c7eeea0f822 100644 --- a/sdk/python/feast/infra/registry/proto_registry_utils.py +++ b/sdk/python/feast/infra/registry/proto_registry_utils.py @@ -214,7 +214,7 @@ def list_saved_datasets( ) -> List[SavedDataset]: saved_datasets = [] for saved_dataset in registry_proto.saved_datasets: - if saved_dataset.project == project: + if saved_dataset.spec.project == project: saved_datasets.append(SavedDataset.from_proto(saved_dataset)) return saved_datasets