From 9fa06c966358e3f157870a7176ada731eb3575b6 Mon Sep 17 00:00:00 2001
From: filtered <176114999+webfiltered@users.noreply.github.com>
Date: Wed, 29 Jan 2025 00:39:53 +1100
Subject: [PATCH] Fix test state polution.

---
 tests-unit/comfy_test/folder_path_test.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/tests-unit/comfy_test/folder_path_test.py b/tests-unit/comfy_test/folder_path_test.py
index 481a59404..775e15c36 100644
--- a/tests-unit/comfy_test/folder_path_test.py
+++ b/tests-unit/comfy_test/folder_path_test.py
@@ -32,7 +32,11 @@ def set_base_dir():
         with patch.object(sys, 'argv', ["main.py", "--base-directory", base_dir]):
             reload(comfy.cli_args)
             reload(folder_paths)
-    return _set_base_dir
+    yield _set_base_dir
+    # Reload the modules after each test to ensure isolation
+    with patch.object(sys, 'argv', ["main.py"]):
+        reload(comfy.cli_args)
+        reload(folder_paths)
 
 
 def test_get_directory_by_type(clear_folder_paths):