comfy-env 0.0.38__tar.gz → 0.0.40__tar.gz

{comfy_env-0.0.38 → comfy_env-0.0.40}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comfy-env
-Version: 0.0.38
+Version: 0.0.40
 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.38 → comfy_env-0.0.40}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "comfy-env"
-version = "0.0.38"
+version = "0.0.40"
 description = "Environment management for ComfyUI custom nodes - CUDA wheel resolution and process isolation"
 readme = "README.md"
 license = {text = "MIT"}

{comfy_env-0.0.38 → comfy_env-0.0.40}/src/comfy_env/__init__.py

@@ -31,7 +31,7 @@ from .ipc.worker import BaseWorker, register
 from .decorator import isolated, shutdown_all_processes
 
 # New in-place installation API
-from .install import install, verify_installation
+from .install import install, verify_installation, setup
 from .resolver import RuntimeEnv, WheelResolver
 
 # Pixi integration (for conda packages)

{comfy_env-0.0.38 → comfy_env-0.0.40}/src/comfy_env/install.py

@@ -18,6 +18,7 @@ Example:
     install(config="comfy-env.toml", mode="isolated")
 """
 
+import inspect
 import shutil
 import subprocess
 import sys
@@ -828,3 +829,42 @@ def verify_installation(
             all_ok = False
 
     return all_ok
+
+
+def setup(
+    log_callback: Optional[Callable[[str], None]] = None,
+    dry_run: bool = False,
+) -> bool:
+    """
+    One-liner setup that auto-discovers config from caller's directory.
+
+    This is the simplest way to install dependencies - just call setup()
+    from your install.py and it will find the comfy-env.toml in the same
+    directory as the calling script.
+
+    Example:
+        # install.py (entire file)
+        from comfy_env import setup
+        setup()
+
+    Args:
+        log_callback: Optional callback for logging. Defaults to print.
+        dry_run: If True, show what would be installed without installing.
+
+    Returns:
+        True if installation succeeded.
+
+    Raises:
+        FileNotFoundError: If no config file found.
+        InstallError: If installation fails.
+    """
+    # Get the caller's directory by inspecting the stack
+    frame = inspect.stack()[1]
+    caller_file = frame.filename
+    caller_dir = Path(caller_file).parent.resolve()
+
+    return install(
+        node_dir=caller_dir,
+        log_callback=log_callback,
+        dry_run=dry_run,
+    )

comfy_env-0.0.40/src/comfy_env/stubs/comfy/__init__.py

@@ -0,0 +1,6 @@
+"""
+Comfy stubs package for isolated worker processes.
+
+Provides minimal implementations of commonly used comfy modules
+without requiring the full ComfyUI installation.
+"""

comfy_env-0.0.40/src/comfy_env/stubs/comfy/model_management.py

@@ -0,0 +1,58 @@
+"""
+Stub for comfy.model_management in isolated worker processes.
+
+Provides device detection and memory management functions without
+requiring the full ComfyUI installation.
+"""
+
+import torch
+
+
+def get_torch_device():
+    """Return the best available torch device."""
+    if torch.cuda.is_available():
+        return torch.device("cuda")
+    elif hasattr(torch.backends, "mps") and torch.backends.mps.is_available():
+        return torch.device("mps")
+    return torch.device("cpu")
+
+
+def get_free_memory(device=None, torch_free_too=False):
+    """Return free VRAM in bytes."""
+    if device is None:
+        device = get_torch_device()
+    if device.type == "cuda":
+        free, total = torch.cuda.mem_get_info(device)
+        return free
+    return 0
+
+
+def get_total_memory(device=None, torch_total_too=False):
+    """Return total VRAM in bytes."""
+    if device is None:
+        device = get_torch_device()
+    if device.type == "cuda":
+        free, total = torch.cuda.mem_get_info(device)
+        return total
+    return 0
+
+
+def soft_empty_cache(force=False):
+    """Clear CUDA cache."""
+    if torch.cuda.is_available():
+        torch.cuda.empty_cache()
+
+
+def unload_all_models():
+    """No-op in isolated worker - models managed by the node itself."""
+    pass
+
+
+def interrupt_current_processing(value=True):
+    """No-op in isolated worker."""
+    pass
+
+
+def processing_interrupted():
+    """Always returns False in isolated worker."""
+    return False

comfy_env-0.0.40/src/comfy_env/stubs/comfy/utils.py

@@ -0,0 +1,29 @@
+"""
+Stub for comfy.utils in isolated worker processes.
+
+Provides utility classes like ProgressBar without requiring
+the full ComfyUI installation.
+"""
+
+
+class ProgressBar:
+    """
+    No-op progress bar for isolated workers.
+
+    In isolated subprocess, we can't update the main ComfyUI progress bar,
+    so this just tracks progress internally. Nodes can still use the same API.
+    """
+
+    def __init__(self, total):
+        self.total = total
+        self.current = 0
+
+    def update(self, value):
+        """Increment progress by value."""
+        self.current += value
+
+    def update_absolute(self, value, total=None, preview=None):
+        """Set progress to absolute value."""
+        self.current = value
+        if total is not None:
+            self.total = total