fix: Teardown tables for Snowflake Materialization testing

Signed-off-by: Miles Adkins <miles.adkins@snowflake.com>
feast-dev · Aug 19, 2022 · 3408181 · 3408181
1 parent f3b522b
commit 3408181
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sdk/python/tests/integration/materialization/test_snowflake.py b/sdk/python/tests/integration/materialization/test_snowflake.py
@@ -81,6 +81,7 @@ def test_snowflake_materialization_consistency_internal():
         validate_offline_online_store_consistency(fs, driver_stats_fv, split_dt)
     finally:
         fs.teardown()
+        snowflake_environment.data_source_creator.teardown()
 
 
 @pytest.mark.integration
@@ -123,3 +124,4 @@ def test_snowflake_materialization_consistency_external():
         validate_offline_online_store_consistency(fs, driver_stats_fv, split_dt)
     finally:
         fs.teardown()
+        snowflake_environment.data_source_creator.teardown()