[fix] IsADirectoryError

bigdata-ustc · Mar 23, 2021 · bba36d3 · bba36d3
1 parent 22cc206
commit bba36d3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/mcd/test_mcd.py b/tests/mcd/test_mcd.py
@@ -7,5 +7,6 @@ def test_train(data, conf, tmp_path):
     user_num, item_num = conf
     cdm = MCD(user_num, item_num, 10)
     cdm.train(data, test_data=data, epoch=2)
-    cdm.save(tmp_path)
-    cdm.load(tmp_path)
+    filepath = tmp_path / "mcd.params"
+    cdm.save(filepath)
+    cdm.load(filepath)