comfy-env 0.0.43__tar.gz → 0.0.44__tar.gz

{comfy_env-0.0.43 → comfy_env-0.0.44}/PKG-INFO

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

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

{comfy_env-0.0.43 → comfy_env-0.0.44}/src/comfy_env/env/manager.py

@@ -383,13 +383,19 @@ class IsolatedEnvManager:
 
                 if "wheel_template" in config:
                     # Direct wheel URL from template
-                    effective_version = version or config.get("default_version")
-                    if not effective_version:
-                        raise RuntimeError(f"Package {package} requires version (no default in registry)")
-
-                    vars_dict["version"] = effective_version
-                    wheel_url = self._substitute_template(config["wheel_template"], vars_dict)
-                    self.log(f"  Installing {package}=={effective_version}...")
+                    template = config["wheel_template"]
+
+                    # Only require version if template uses {version}
+                    effective_version = None
+                    if "{version}" in template:
+                        effective_version = version or config.get("default_version")
+                        if not effective_version:
+                            raise RuntimeError(f"Package {package} requires version (no default in registry)")
+                        vars_dict["version"] = effective_version
+
+                    wheel_url = self._substitute_template(template, vars_dict)
+                    version_str = f"=={effective_version}" if effective_version else ""
+                    self.log(f"  Installing {package}{version_str}...")
                     self.log(f"    URL: {wheel_url}")
                     result = subprocess.run(
                         pip_args + ["--no-deps", wheel_url],
@@ -398,6 +404,19 @@ class IsolatedEnvManager:
                     if result.returncode != 0:
                         raise RuntimeError(f"Failed to install {package}: {result.stderr}")
 
+                elif "find_links" in config:
+                    # find_links: pip resolves wheel from index URL
+                    find_links_url = config["find_links"]
+                    effective_version = version or config.get("default_version")
+                    pkg_spec = f"{package}=={effective_version}" if effective_version else package
+                    self.log(f"  Installing {pkg_spec} from {find_links_url}...")
+                    result = subprocess.run(
+                        pip_args + ["--no-deps", "--find-links", find_links_url, pkg_spec],
+                        capture_output=True, text=True,
+                    )
+                    if result.returncode != 0:
+                        raise RuntimeError(f"Failed to install {package}: {result.stderr}")
+
                 elif "package_name" in config:
                     # PyPI variant (e.g., spconv-cu124)
                     pkg_name = self._substitute_template(config["package_name"], vars_dict)
@@ -411,7 +430,7 @@ class IsolatedEnvManager:
                         raise RuntimeError(f"Failed to install {package}: {result.stderr}")
 
                 else:
-                    raise RuntimeError(f"Package {package} in registry but missing wheel_template or package_name")
+                    raise RuntimeError(f"Package {package} in registry but missing wheel_template, find_links, or package_name")
 
             else:
                 # Not in registry - error

{comfy_env-0.0.43 → comfy_env-0.0.44}/src/comfy_env/install.py

@@ -367,11 +367,25 @@ def _resolve_wheel_url(
 
         # wheel_template: direct URL
         if "wheel_template" in config:
+            template = config["wheel_template"]
+
+            # Only require version if template uses {version}
+            if "{version}" in template:
+                effective_version = version or config.get("default_version")
+                if not effective_version:
+                    raise InstallError(f"Package {package} requires version (no default in registry)")
+                vars_dict["version"] = effective_version
+
+            return _substitute_template(template, vars_dict)
+
+        # find_links: pip resolves wheel from index (e.g., miropsota's detectron2)
+        if "find_links" in config:
+            find_links_url = config["find_links"]
             effective_version = version or config.get("default_version")
-            if not effective_version:
-                raise InstallError(f"Package {package} requires version (no default in registry)")
-            vars_dict["version"] = effective_version
-            return _substitute_template(config["wheel_template"], vars_dict)
+            if effective_version:
+                return f"find_links:{find_links_url}:{package}=={effective_version}"
+            else:
+                return f"find_links:{find_links_url}:{package}"
 
         # package_name: PyPI variant (e.g., spconv-cu124)
         if "package_name" in config:
@@ -406,6 +420,11 @@ def _install_cuda_package(
         pkg_spec = f"{pkg_name}=={version}" if version else pkg_name
         log(f"  Installing {package} as {pkg_spec} from PyPI...")
         _pip_install([pkg_spec], no_deps=False, log=log)
+    elif url_or_marker.startswith("find_links:"):
+        # find_links: pip resolves from index URL
+        _, find_links_url, pkg_spec = url_or_marker.split(":", 2)
+        log(f"  Installing {pkg_spec} from {find_links_url}...")
+        _pip_install([pkg_spec, "--find-links", find_links_url], no_deps=True, log=log)
     else:
         # Direct wheel URL
         log(f"  Installing {package}...")

{comfy_env-0.0.43 → comfy_env-0.0.44}/src/comfy_env/wheel_sources.yml

@@ -25,18 +25,22 @@ packages:
   # ===========================================================================
   torch-scatter:
     wheel_template: "https://data.pyg.org/whl/torch-{torch_version}%2Bcu{cuda_short}/torch_scatter-{version}%2Bpt{torch_mm}cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "2.1.2"
     description: Scatter operations for PyTorch
 
   torch-cluster:
     wheel_template: "https://data.pyg.org/whl/torch-{torch_version}%2Bcu{cuda_short}/torch_cluster-{version}%2Bpt{torch_mm}cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "1.6.3"
     description: Clustering algorithms for PyTorch
 
   torch-sparse:
     wheel_template: "https://data.pyg.org/whl/torch-{torch_version}%2Bcu{cuda_short}/torch_sparse-{version}%2Bpt{torch_mm}cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "0.6.18"
     description: Sparse tensor operations for PyTorch
 
   torch-spline-conv:
     wheel_template: "https://data.pyg.org/whl/torch-{torch_version}%2Bcu{cuda_short}/torch_spline_conv-{version}%2Bpt{torch_mm}cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "1.2.2"
     description: Spline convolutions for PyTorch
 
   # ===========================================================================
@@ -114,7 +118,8 @@ packages:
   # Prebuilt wheels from miropsota's torch_packages_builder
   # ===========================================================================
   detectron2:
-    wheel_template: "https://miropsota.github.io/torch_packages_builder/detectron2/detectron2-0.6%2B2a420edpt{torch_version}cu{cuda_short}-{py_tag}-{py_tag}-{platform}.whl"
+    find_links: "https://miropsota.github.io/torch_packages_builder/detectron2/"
+    default_version: "0.6"
     description: Detectron2 - Facebook's detection and segmentation library
 
   # ===========================================================================
@@ -131,5 +136,6 @@ packages:
   # Note: This uses a special package_name field, not wheel_template
   # ===========================================================================
   spconv:
-    package_name: "spconv-cu{cuda_short2}"
-    description: Sparse convolution library (installs spconv-cu124, spconv-cu126, etc.)
+    wheel_template: "https://github.com/PozzettiAndrea/cuda-wheels/releases/download/spconv_cu{cuda_short}-latest/spconv-{version}%2Bcu{cuda_short}torch{torch_mm}-{py_tag}-{py_tag}-{platform}.whl"
+    default_version: "2.3.8"
+    description: Sparse convolution library