only test for stable output

thuml · Nov 28, 2023 · c4a67e9 · c4a67e9
1 parent 7b8b727
commit c4a67e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_pytorch/cmp_output.py b/tests/test_pytorch/cmp_output.py
@@ -1,9 +1,9 @@
 import glob
 
-output_files = glob.glob("depyf_output/*/*.py")
+output_files = glob.glob("depyf_output/*/__*.py")
 output_files.sort()
 
-expected_files = glob.glob("tests/depyf_output/*/*.py")
+expected_files = glob.glob("tests/depyf_output/*/__*.py")
 expected_files.sort()
 
 assert len(output_files) == len(expected_files), f"len(output_files)={len(output_files)}, len(expected_files)={len(expected_files)}"