From 689db36073ce562f576c5b7aed082aceb5f152bf Mon Sep 17 00:00:00 2001 From: Jacob Segal Date: Fri, 25 Jul 2025 14:32:27 -0700 Subject: [PATCH] Remove the need for `--generate-api-stubs` --- comfy/cli_args.py | 1 - comfy_api/generate_api_stubs.py | 2 +- main.py | 6 ------ 3 files changed, 1 insertion(+), 8 deletions(-) diff --git a/comfy/cli_args.py b/comfy/cli_args.py index b3bf0fc62..7234a7ba0 100644 --- a/comfy/cli_args.py +++ b/comfy/cli_args.py @@ -153,7 +153,6 @@ parser.add_argument("--disable-metadata", action="store_true", help="Disable sav parser.add_argument("--disable-all-custom-nodes", action="store_true", help="Disable loading all custom nodes.") parser.add_argument("--whitelist-custom-nodes", type=str, nargs='+', default=[], help="Specify custom node folders to load even when --disable-all-custom-nodes is enabled.") parser.add_argument("--disable-api-nodes", action="store_true", help="Disable loading all api nodes.") -parser.add_argument("--generate-api-stubs", action="store_true", help="Generate .pyi stub files for API sync wrappers and exit.") parser.add_argument("--multi-user", action="store_true", help="Enables per-user storage.") diff --git a/comfy_api/generate_api_stubs.py b/comfy_api/generate_api_stubs.py index 9d300e2db..604a7eced 100644 --- a/comfy_api/generate_api_stubs.py +++ b/comfy_api/generate_api_stubs.py @@ -10,7 +10,7 @@ import logging import importlib # Add ComfyUI to path so we can import modules -sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) +sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from comfy_api.internal.async_to_sync import AsyncToSyncConverter from comfy_api.version_list import supported_versions diff --git a/main.py b/main.py index 9217ea8f3..86319f1a7 100644 --- a/main.py +++ b/main.py @@ -22,12 +22,6 @@ if __name__ == "__main__": setup_logger(log_level=args.verbose, use_stdout=args.log_stdout) -# Handle --generate-api-stubs early -if args.generate_api_stubs: - from comfy_api.generate_api_stubs import main as generate_stubs_main - generate_stubs_main() - sys.exit(0) - def apply_custom_paths(): # extra model paths extra_model_paths_config_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "extra_model_paths.yaml")