Skip to content

Commit 4e09b3d

Browse files
authored
touch models in the tests
Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
1 parent 7788431 commit 4e09b3d

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/simplify/test_run_model_devi.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,9 @@ class TestOneH5(unittest.TestCase):
1717
def setUp(self):
1818
work_path = Path("iter.000000") / "01.model_devi"
1919
work_path.mkdir(parents=True, exist_ok=True)
20+
# fake models
21+
for ii in range(4):
22+
(work_path / f"graph.{ii:03d}.pb").torch()
2023
with tempfile.TemporaryDirectory() as tmpdir:
2124
with open(Path(tmpdir) / "test.xyz", "w") as f:
2225
f.write(

0 commit comments

Comments
 (0)