fix test

Sandy4321 · May 10, 2023 · dc0d593 · dc0d593
1 parent c9c997f
commit dc0d593
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pygod/test/test_gae.py b/pygod/test/test_gae.py
@@ -20,7 +20,7 @@ def setUp(self):
         # the roc should be higher than this; it is model dependent
         self.roc_floor = 0.55
 
-        test_graph = torch.load(os.path.join('pygod/test/test_graph.pt'))
+        test_graph = torch.load(os.path.join('pygod/test/train_graph.pt'))
         self.data = test_graph
 
         self.model = GAE(epoch=5)

diff --git a/pygod/test/test_utility.py b/pygod/test/test_utility.py
@@ -142,5 +142,5 @@ def test_is_fitted(self):
         with assert_raises(Exception):
             is_fitted(detector)
 
-        detector.fit(torch.load('test_graph.pt'))
+        detector.fit(torch.load('pygod.test.test_graph.pt'))
         is_fitted(detector)