From c4a67e990dd0abe1d2ceec82884379f7b44f14ea Mon Sep 17 00:00:00 2001
From: youkaichao <youkaichao@126.com>
Date: Tue, 28 Nov 2023 17:36:46 +0800
Subject: [PATCH] only test for stable output

---
 tests/test_pytorch/cmp_output.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_pytorch/cmp_output.py b/tests/test_pytorch/cmp_output.py
index 38581167..4c4abfa7 100644
--- 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)}"