From 9694b84c9cb627cace87c5102372a5da5e94c906 Mon Sep 17 00:00:00 2001
From: atmaranto <35581227+atmaranto@users.noreply.github.com>
Date: Wed, 15 May 2024 05:11:54 -0500
Subject: [PATCH] Don't alter sys.argv when imported

---
 comfyui_to_python.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfyui_to_python.py b/comfyui_to_python.py
index fe01d8d..06ef5c2 100644
--- a/comfyui_to_python.py
+++ b/comfyui_to_python.py
@@ -339,7 +339,7 @@ class CodeGenerator:
         argparse_code.append(f'parser.add_argument("--disable-metadata", action="store_true", help="Disables writing workflow metadata to the outputs")\n')
         argparse_code.append('''
 comfy_args = [sys.argv[0]]
-if "--" in sys.argv:
+if __name__ == "__main__" and "--" in sys.argv:
     idx = sys.argv.index("--")
     comfy_args += sys.argv[idx+1:]
     sys.argv = sys.argv[:idx]