From 5f8160540ad5b34c640e3d87aa2c48f5f1b3087c Mon Sep 17 00:00:00 2001 From: Xiaoyong Zhu Date: Wed, 18 May 2022 12:02:59 -0700 Subject: [PATCH] fix test failure --- feathr_project/test/test_azure_snowflake_e2e.py | 2 +- feathr_project/test/test_feature_registry.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/feathr_project/test/test_azure_snowflake_e2e.py b/feathr_project/test/test_azure_snowflake_e2e.py index 1bc3bd0cd..b1f6785ca 100644 --- a/feathr_project/test/test_azure_snowflake_e2e.py +++ b/feathr_project/test/test_azure_snowflake_e2e.py @@ -41,7 +41,7 @@ def test_feathr_online_store_agg_features(): assert len(res) == 2 assert res[0] != None assert res[1] != None - res = client.multi_get_online_features('snowflakeSampleDemoFeature', + res = client.multi_get_online_features(online_test_table, ['1', '2'], ['f_snowflake_call_center_division_name', 'f_snowflake_call_center_zipcode']) assert res['1'][0] != None diff --git a/feathr_project/test/test_feature_registry.py b/feathr_project/test/test_feature_registry.py index 4abb73b6d..20eedfb6d 100644 --- a/feathr_project/test/test_feature_registry.py +++ b/feathr_project/test/test_feature_registry.py @@ -60,7 +60,7 @@ def test_feathr_register_features_e2e(): client.get_offline_features(observation_settings=settings, feature_query=feature_query, output_path=output_path) - client.wait_job_to_finish(timeout_sec=500) + client.wait_job_to_finish(timeout_sec=900)