Skip to content

Commit 0fe2df8

Browse files
committed
fix
Signed-off-by: Danny Chiao <[email protected]>
1 parent e17973f commit 0fe2df8

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

sdk/python/feast/diff/registry_diff.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -81,10 +81,11 @@ def to_string(self):
8181
def tag_objects_for_keep_delete_update_add(
8282
existing_objs: Iterable[FeastObject], desired_objs: Iterable[FeastObject]
8383
) -> Tuple[List[FeastObject], List[FeastObject], List[FeastObject], List[FeastObject]]:
84-
existing_obj_names = {e.name for e in existing_objs}
85-
desired_objs = list(desired_objs)
86-
existing_objs = list(existing_objs)
87-
desired_obj_names = {e.name for e in desired_objs}
84+
# TODO(adchia): Remove the "if X.name" condition when data sources are forced to have names
85+
existing_obj_names = {e.name for e in existing_objs if e.name}
86+
desired_objs = [obj for obj in desired_objs if obj.name]
87+
existing_objs = [obj for obj in existing_objs if obj.name]
88+
desired_obj_names = {e.name for e in desired_objs if e.name}
8889

8990
objs_to_add = [e for e in desired_objs if e.name not in existing_obj_names]
9091
objs_to_update = [e for e in desired_objs if e.name in existing_obj_names]

sdk/python/tests/integration/registration/test_cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ def test_universal_cli(environment: Environment):
8989
["data-sources", "describe", "customer_profile_source"], cwd=repo_path,
9090
)
9191
assertpy.assert_that(result.returncode).is_equal_to(0)
92-
assertpy.assert_that(fs.list_data_sources()).is_length(5)
92+
assertpy.assert_that(fs.list_data_sources()).is_length(4)
9393

9494
# entity & feature view describe commands should fail when objects don't exist
9595
result = runner.run(["entities", "describe", "foo"], cwd=repo_path)

0 commit comments

Comments
 (0)