From 24068d568315b697bf223eaafd0d375a8c2c70a4 Mon Sep 17 00:00:00 2001 From: Breno Costa Date: Wed, 15 May 2024 14:57:35 +0200 Subject: [PATCH] fixup! Reuse same setup and assertion --- .../integration/online_store/test_universal_online.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/sdk/python/tests/integration/online_store/test_universal_online.py b/sdk/python/tests/integration/online_store/test_universal_online.py index 7944a1be106..fc4760aeb07 100644 --- a/sdk/python/tests/integration/online_store/test_universal_online.py +++ b/sdk/python/tests/integration/online_store/test_universal_online.py @@ -402,7 +402,9 @@ def test_online_retrieval_with_shared_batch_source(environment, universal_data_s ) -def setup_feature_store_universal_feature_views(environment, universal_data_sources) -> FeatureStore: +def setup_feature_store_universal_feature_views( + environment, universal_data_sources +) -> FeatureStore: fs: FeatureStore = environment.feature_store entities, datasets, data_sources = universal_data_sources feature_views = construct_universal_feature_views(data_sources) @@ -459,7 +461,7 @@ def assert_feature_store_universal_feature_views_response(df: pd.DataFrame): def test_online_retrieval_with_event_timestamps( environment, universal_data_sources, full_feature_names ): - fs = setup_feature_store_universal_feature_views(fs, universal_data_sources) + fs = setup_feature_store_universal_feature_views(universal_data_sources) response = fs.get_online_features( features=[ @@ -479,7 +481,7 @@ def test_online_retrieval_with_event_timestamps( def test_async_online_retrieval_with_event_timestamps( environment, universal_data_sources ): - fs = setup_feature_store_universal_feature_views(fs, universal_data_sources) + fs = setup_feature_store_universal_feature_views(universal_data_sources) response = asyncio.run( fs.get_online_features_async(