clean-up after test

zenml-io · Oct 19, 2023 · 3605a88 · 3605a88
1 parent 681feb3
commit 3605a88
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_template.py b/tests/test_template.py
@@ -72,6 +72,7 @@ def generate_and_run_project(
         dst_path=str(dst_path),
         data=answers,
         unsafe=True,
+        vcs_ref="HEAD",
     ) as worker:
         worker.run_copy()
 
@@ -104,6 +105,8 @@ def generate_and_run_project(
 
             # clean up
             Client().delete_pipeline(product_name + pipeline_suffix)
+        Client().delete_model(product_name)
+        Client().active_stack.model_registry.delete_model(product_name)
 
     os.chdir(current_dir)
     shutil.rmtree(dst_path)