From b536cdb566bcad4315aa2d323266ac2db84b0519 Mon Sep 17 00:00:00 2001 From: Malcolm Keyes <78397791+malcolmk181@users.noreply.github.com> Date: Thu, 7 Sep 2023 01:51:00 -0400 Subject: [PATCH] fix: Saved datasets no longer break CLI registry-dump command (#3717) 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