Fix Create Files Unit Test (#1668)

* ignore __pycache__ when checking for new files * formatting fix
neuralmagic · Jul 13, 2023 · 6d836ed · 6d836ed
1 parent 3593b1a
commit 6d836ed
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/sparseml/conftest.py b/tests/sparseml/conftest.py
@@ -49,7 +49,10 @@ def check_for_created_files():
     if os.path.isdir(log_dir):
         shutil.rmtree(log_dir)
 
-    end_files_root = _get_files(directory=r".")
+    # allow creation of __pycache__ directories
+    end_files_root = [
+        f_path for f_path in _get_files(directory=r".") if "__pycache__" not in f_path
+    ]
     end_files_temp = _get_files(directory=tempfile.gettempdir())
 
     # assert no files created in root directory while running