fix: Teardown tables for Snowflake Materialization testing (feast-dev#3106)

sfc-gh-madkins · web-flow · commit 0a0c974f38e7 · 2022-08-18T20:44:37.000-07:00
Signed-off-by: Miles Adkins &lt;miles.adkins@snowflake.com&gt;

Signed-off-by: Miles Adkins &lt;miles.adkins@snowflake.com&gt;
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()
