making test_multihit output ignored

matsengrp · Sep 27, 2024 · cbd8484 · cbd8484
1 parent 225a7c6
commit cbd8484
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/test_multihit.py b/tests/test_multihit.py
@@ -1,3 +1,5 @@
+import os
+
 import netam.multihit as multihit
 import netam.framework as framework
 import netam.hit_class as hit_class
@@ -81,8 +83,9 @@ def test_train(hitclass_burrito):
 
 
 def test_serialize(hitclass_burrito):
-    hitclass_burrito.save_crepe("test_multihit_crepe")
-    new_crepe = framework.load_crepe("test_multihit_crepe")
+    os.makedirs("_ignore", exist_ok=True)
+    hitclass_burrito.save_crepe("_ignore/test_multihit_crepe")
+    new_crepe = framework.load_crepe("_ignore/test_multihit_crepe")
     assert torch.allclose(new_crepe.model.values, hitclass_burrito.model.values)