comfy-env 0.0.37__tar.gz → 0.0.39__tar.gz

{comfy_env-0.0.37 → comfy_env-0.0.39}/PKG-INFO

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

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

{comfy_env-0.0.37 → comfy_env-0.0.39}/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.37 → comfy_env-0.0.39}/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.37 → comfy_env-0.0.39}/src/comfy_env/wheel_sources.yml

@@ -49,13 +49,13 @@ packages:
     description: Spline convolutions for PyTorch
 
   # ===========================================================================
-  # pytorch3d - MiroPsota's torch_packages_builder
-  # https://github.com/MiroPsota/torch_packages_builder
+  # pytorch3d - PyTorch3D 3D deep learning library
+  # https://pozzettiandrea.github.io/cuda-wheels
   # ===========================================================================
   pytorch3d:
     method: index
-    index_url: "https://miropsota.github.io/torch_packages_builder/pytorch3d/"
-    wheel_template: "https://miropsota.github.io/torch_packages_builder/pytorch3d/pytorch3d-{version}%2Bpt{torch_version}cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl"
+    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
+    wheel_template: "https://pozzettiandrea.github.io/cuda-wheels/pytorch3d/pytorch3d-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
     default_version: "0.7.9"
     description: PyTorch3D - 3D deep learning library
 
@@ -76,6 +76,11 @@ packages:
     wheel_template: "https://pozzettiandrea.github.io/cuda-wheels/cumesh/cumesh-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
     description: CUDA-accelerated mesh utilities
 
+  cubvh:
+    method: index
+    index_url: "https://pozzettiandrea.github.io/cuda-wheels"
+    description: CUDA BVH acceleration for mesh operations and marching cubes
+
   o_voxel:
     method: index
     index_url: "https://pozzettiandrea.github.io/cuda-wheels"