comfy-env 0.0.9__py3-none-any.whl → 0.0.11__py3-none-any.whl

comfy_env/__init__.py

@@ -40,7 +40,7 @@ This package provides:
 The @isolated decorator and WorkerBridge are still available.
 """
 
-__version__ = "0.0.8"
+__version__ = "0.0.11"
 
 from .env.config import IsolatedEnv, EnvManagerConfig, LocalConfig, NodeReq
 from .env.config_file import (

comfy_env/env/detection.py

@@ -7,10 +7,16 @@ vs older GPUs which use CUDA 12.4.
 This runs BEFORE PyTorch is installed, so we use nvidia-smi directly.
 """
 
+import os
 import subprocess
 from typing import List, Dict, Optional
 
 
+# Environment variable to override CUDA version detection
+# Useful for CI environments without GPU
+CUDA_VERSION_ENV_VAR = "COMFY_ENV_CUDA_VERSION"
+
+
 def detect_gpu_info() -> List[Dict[str, str]]:
     """
     Detect GPU name and compute capability using nvidia-smi.
@@ -126,7 +132,12 @@ def detect_cuda_version() -> Optional[str]:
     Returns:
         "12.4" for Pascal or older (compute < 7.5),
         "12.8" for Turing or newer (compute >= 7.5),
-        None if no GPU detected.
+        None if no GPU detected and no env var override.
+
+    Environment Variable Override:
+        Set COMFY_ENV_CUDA_VERSION to override auto-detection.
+        Useful for CI environments without GPU.
+        Example: COMFY_ENV_CUDA_VERSION=12.8
 
     GPU Architecture Reference:
         - Pascal (GTX 10xx, P100): compute 6.0-6.1 → CUDA 12.4
@@ -138,6 +149,15 @@ def detect_cuda_version() -> Optional[str]:
     """
     gpus = detect_gpu_info()
     if not gpus:
+        # No GPU detected - check for env var override
+        override = os.environ.get(CUDA_VERSION_ENV_VAR)
+        if override:
+            # Normalize format (e.g., "12.8" or "128" -> "12.8")
+            override = override.strip()
+            if override and '.' not in override and len(override) >= 2:
+                # Convert "128" -> "12.8"
+                override = f"{override[:-1]}.{override[-1]}"
+            return override
         return None
 
     # Check if any GPU is legacy (Pascal or older)
@@ -159,7 +179,10 @@ def get_gpu_summary() -> str:
     gpus = detect_gpu_info()
 
     if not gpus:
-        return "No NVIDIA GPU detected"
+        override = os.environ.get(CUDA_VERSION_ENV_VAR)
+        if override:
+            return f"No NVIDIA GPU detected (using {CUDA_VERSION_ENV_VAR}={override})"
+        return f"No NVIDIA GPU detected (set {CUDA_VERSION_ENV_VAR} to override)"
 
     lines = []
     for i, gpu in enumerate(gpus):

comfy_env/env/manager.py

@@ -6,6 +6,7 @@ Uses uv for fast environment creation and package installation.
 
 import subprocess
 import shutil
+import sys
 from pathlib import Path
 from typing import Optional, Callable
 

{comfy_env-0.0.9.dist-info → comfy_env-0.0.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comfy-env
-Version: 0.0.9
+Version: 0.0.11
 Summary: Environment management for ComfyUI custom nodes - CUDA wheel resolution and process isolation
 Project-URL: Homepage, https://github.com/PozzettiAndrea/comfy-env
 Project-URL: Repository, https://github.com/PozzettiAndrea/comfy-env

{comfy_env-0.0.9.dist-info → comfy_env-0.0.11.dist-info}/RECORD

@@ -1,4 +1,4 @@
-comfy_env/__init__.py,sha256=U5XtB2UeXZi8JPBhlh0XjQprLsAuVslPoJNWOSvCFhs,4249
+comfy_env/__init__.py,sha256=1EjSpAK20Z6RAXKn3onZwj2a_FIctG0E_LhWlE7SgIs,4250
 comfy_env/cli.py,sha256=9GvQYrXlJRl0ZaCuFHvRtVxWQ34Axd5Brgu5FWRONp4,11424
 comfy_env/decorator.py,sha256=daFR5aLzshkmo5sRKhSGPcTUgIUWml7Gs6A1bfnDuyc,15718
 comfy_env/errors.py,sha256=egeyXY-j7KpxyA0s67TcJLEfJX23LCAD3v1P4FgQIGE,10917
@@ -9,8 +9,8 @@ comfy_env/runner.py,sha256=0YUqzK93u--7pKV6_yVC564AJE9rS3y81t5ZhQi2t4Y,9696
 comfy_env/env/__init__.py,sha256=sybOBrxJCfL4Xry9NNd5xwn9hXIHudXlXDa7SpJkPCE,811
 comfy_env/env/config.py,sha256=R8JyE5iQLHKgnxXOGA8SAI7iu2eYSfXn-MsaqHoU2_A,5667
 comfy_env/env/config_file.py,sha256=2sTeBUqdDYPwQfUO-3Mu2zTsgxQ4IhllRxEd2a4pgPg,20593
-comfy_env/env/detection.py,sha256=Co8BJmTRCq1ZHDsm6832jF87za0GRAhH7zF04-5QwcE,4949
-comfy_env/env/manager.py,sha256=hGaK9C_nRwfHthowgFv2rPhhpfAMDJhxPEdQFFv6BbI,24770
+comfy_env/env/detection.py,sha256=Br_Ed2aqGR6_M01UMZnrzVky3a2iyzz1XuhOXduM1zw,5993
+comfy_env/env/manager.py,sha256=MRmqg3Qw0IYUwR6FtM3ZIcGZvWcnKDyO8xbzIAxnwCk,24782
 comfy_env/env/security.py,sha256=dNSitAnfBNVdvxgBBntYw33AJaCs_S1MHb7KJhAVYzM,8171
 comfy_env/env/platform/__init__.py,sha256=Nb5MPZIEeanSMEWwqU4p4bnEKTJn1tWcwobnhq9x9IY,614
 comfy_env/env/platform/base.py,sha256=iS0ptTTVjXRwPU4qWUdvHI7jteuzxGSjWr5BUQ7hGiU,2453
@@ -32,8 +32,8 @@ comfy_env/workers/pool.py,sha256=MtjeOWfvHSCockq8j1gfnxIl-t01GSB79T5N4YB82Lg,695
 comfy_env/workers/tensor_utils.py,sha256=TCuOAjJymrSbkgfyvcKtQ_KbVWTqSwP9VH_bCaFLLq8,6409
 comfy_env/workers/torch_mp.py,sha256=DsfxE3LBAWEuGtk-p-YL0UhVQ7VDh73KT_TFRxYN4-Q,12563
 comfy_env/workers/venv.py,sha256=_ekHfZPqBIPY08DjqiXm6cTBQH4DrbxRWR3AAv3mit8,31589
-comfy_env-0.0.9.dist-info/METADATA,sha256=HwKjPaZyiGGsJ47w8nempK8MokAu1YbWS1ZqwSCXS_4,5371
-comfy_env-0.0.9.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-comfy_env-0.0.9.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
-comfy_env-0.0.9.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
-comfy_env-0.0.9.dist-info/RECORD,,
+comfy_env-0.0.11.dist-info/METADATA,sha256=oTBqvxpgXd5ctKduTg7654WXVNLkksrdpAoxuPbr0aY,5372
+comfy_env-0.0.11.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+comfy_env-0.0.11.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
+comfy_env-0.0.11.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
+comfy_env-0.0.11.dist-info/RECORD,,