Update test_run_model_devi.py

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Jul 18, 2024 · d22f754 · d22f754
1 parent 4e09b3d
commit d22f754
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/simplify/test_run_model_devi.py b/tests/simplify/test_run_model_devi.py
@@ -19,7 +19,7 @@ def setUp(self):
         work_path.mkdir(parents=True, exist_ok=True)
         # fake models
         for ii in range(4):
-            (work_path / f"graph.{ii:03d}.pb").torch()
+            (work_path / f"graph.{ii:03d}.pb").touch()
         with tempfile.TemporaryDirectory() as tmpdir:
             with open(Path(tmpdir) / "test.xyz", "w") as f:
                 f.write(