fix test

etna-team · Aug 29, 2024 · c4a7adc · c4a7adc
1 parent 14de6c2
commit c4a7adc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tests/test_transforms/test_embeddings/test_models/test_ts2vec.py b/tests/test_transforms/test_embeddings/test_models/test_ts2vec.py
@@ -153,6 +153,7 @@ def test_correct_list_models():
 @pytest.mark.parametrize("model_name", ["ts2vec_tiny"])
 def test_load_pretrained_model_default_path(model_name):
     path = Path.home() / ".etna" / "embeddings" / "ts2vec" / f"{model_name}.zip"
+    path.unlink(missing_ok=True)
     _ = TS2VecEmbeddingModel.load(model_name=model_name)
     assert os.path.isfile(path)
 

diff --git a/tests/test_transforms/test_embeddings/test_models/test_tstcc.py b/tests/test_transforms/test_embeddings/test_models/test_tstcc.py
@@ -159,6 +159,7 @@ def test_correct_list_models():
 @pytest.mark.parametrize("model_name", ["tstcc_medium"])
 def test_load_pretrained_model_default_path(model_name):
     path = Path.home() / ".etna" / "embeddings" / "tstcc" / f"{model_name}.zip"
+    path.unlink(missing_ok=True)
     _ = TSTCCEmbeddingModel.load(path=path, model_name=model_name)
     assert os.path.isfile(path)