diff --git a/sdk/python/tests/unit/local_feast_tests/test_local_feature_store.py b/sdk/python/tests/unit/local_feast_tests/test_local_feature_store.py index 3a4ca796c3..863aaf94e9 100644 --- a/sdk/python/tests/unit/local_feast_tests/test_local_feature_store.py +++ b/sdk/python/tests/unit/local_feast_tests/test_local_feature_store.py @@ -386,7 +386,7 @@ def test_apply_dummy_entity_and_feature_view_columns(test_feature_store): assert fv_with_entity.entity_columns == [] # Register Feature View - test_feature_store.apply([e1, fv_no_entity]) + test_feature_store.apply([e1, fv_no_entity, fv_with_entity]) fv_from_online_store = test_feature_store.get_feature_view( "my_feature_view_no_entity" ) @@ -395,6 +395,7 @@ def test_apply_dummy_entity_and_feature_view_columns(test_feature_store): assert fv_from_online_store.entity_columns[0].name == DUMMY_ENTITY_ID assert fv_from_online_store.entities == [] assert fv_no_entity.entities == [DUMMY_ENTITY_NAME] + assert fv_with_entity.entity_columns[0] == e1.join_key test_feature_store.teardown() diff --git a/sdk/python/tests/unit/test_on_demand_python_transformation.py b/sdk/python/tests/unit/test_on_demand_python_transformation.py index a751e6a024..ce9c6f6672 100644 --- a/sdk/python/tests/unit/test_on_demand_python_transformation.py +++ b/sdk/python/tests/unit/test_on_demand_python_transformation.py @@ -29,8 +29,8 @@ Int64, String, UnixTimestamp, - _utc_now, ValueType, + _utc_now, from_value_type, )