chore: apply refurb suggestions

kraina-ai · Aug 29, 2024 · 5ef3fc7 · 5ef3fc7
1 parent f0b7147
commit 5ef3fc7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/embedders/geovex/test_embedder.py b/tests/embedders/geovex/test_embedder.py
@@ -187,7 +187,7 @@ def test_embedder_save_load() -> None:
         assert isinstance(loaded_embedder._model, GeoVexModel)
 
         # safely clean up tmp_models directory
-        os.remove(tmp_models_dir / "test_model" / "model.pt")
-        os.remove(tmp_models_dir / "test_model" / "config.json")
+        (tmp_models_dir / "test_model" / "model.pt").unlink()
+        (tmp_models_dir / "test_model" / "config.json").unlink()
         os.rmdir(tmp_models_dir / "test_model")
         os.rmdir(tmp_models_dir)