fix: Access saved dataset proto correctly in list_saved_datasets

malcolmk181 · malcolmk181 · commit f6856d47a46a · 2023-08-09T18:21:16.000-04:00
Signed-off-by: Malcolm Keyes &lt;malcolm.a.keyes@gmail.com&gt;
diff --git 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
 
