comfy-env 0.0.32__py3-none-any.whl → 0.0.34__py3-none-any.whl

comfy_env/env/manager.py

@@ -386,20 +386,26 @@ class IsolatedEnvManager:
                 self.log(f"  Installing {package} ({method})...")
 
                 if method == "index":
-                    # PEP 503 index - use --extra-index-url
+                    # PEP 503 index - try to resolve exact wheel URL first
                     index_url = self._substitute_template(config["index_url"], vars_dict)
                     pkg_spec = f"{package}=={version}" if version else package
                     # Try to resolve exact wheel URL from index
                     wheel_url = resolve_wheel_from_index(index_url, package, vars_dict, version)
                     if wheel_url:
+                        # Install from resolved URL directly (guarantees we get what we resolved)
                         self.log(f"    Wheel: {wheel_url}")
+                        result = subprocess.run(
+                            pip_args + ["--no-deps", wheel_url],
+                            capture_output=True, text=True,
+                        )
                     else:
+                        # Fallback to index-based resolution
                         self.log(f"    Index: {index_url}")
                         self.log(f"    Package: {pkg_spec}")
-                    result = subprocess.run(
-                        pip_args + ["--extra-index-url", index_url, "--no-deps", pkg_spec],
-                        capture_output=True, text=True,
-                    )
+                        result = subprocess.run(
+                            pip_args + ["--extra-index-url", index_url, "--no-deps", pkg_spec],
+                            capture_output=True, text=True,
+                        )
 
                 elif method in ("github_index", "find_links"):
                     # GitHub Pages or generic find-links
@@ -408,14 +414,20 @@ class IsolatedEnvManager:
                     # Try to resolve exact wheel URL from find-links page
                     wheel_url = resolve_wheel_from_index(index_url, package, vars_dict, version)
                     if wheel_url:
+                        # Install from resolved URL directly (guarantees we get what we resolved)
                         self.log(f"    Wheel: {wheel_url}")
+                        result = subprocess.run(
+                            pip_args + ["--no-deps", wheel_url],
+                            capture_output=True, text=True,
+                        )
                     else:
+                        # Fallback to find-links based resolution
                         self.log(f"    Find-links: {index_url}")
                         self.log(f"    Package: {pkg_spec}")
-                    result = subprocess.run(
-                        pip_args + ["--find-links", index_url, "--no-deps", pkg_spec],
-                        capture_output=True, text=True,
-                    )
+                        result = subprocess.run(
+                            pip_args + ["--find-links", index_url, "--no-deps", pkg_spec],
+                            capture_output=True, text=True,
+                        )
 
                 elif method == "pypi_variant":
                     # Transform package name based on CUDA version

comfy_env/install.py

@@ -475,7 +475,7 @@ def _install_cuda_package(
         method = config["method"]
 
         if method == "index":
-            # PEP 503 index - use pip --extra-index-url
+            # PEP 503 index - try to resolve exact wheel URL first
             index_url = _substitute_template(config["index_url"], env)
             # Check for version_template (e.g., detectron2 with embedded torch/cuda version)
             if "version_template" in config:
@@ -489,14 +489,17 @@ def _install_cuda_package(
             vars_dict = env.as_dict()
             wheel_url = resolve_wheel_from_index(index_url, package, vars_dict, actual_version)
             if wheel_url:
+                # Install from resolved URL directly (guarantees we get what we resolved)
                 log(f"    Wheel: {wheel_url}")
+                _pip_install([wheel_url], no_deps=True, log=log)
             else:
+                # Fallback to index-based resolution
                 log(f"    Index: {index_url}")
                 log(f"    Package: {pkg_spec}")
-            _pip_install_with_index(pkg_spec, index_url, log)
+                _pip_install_with_index(pkg_spec, index_url, log)
 
         elif method == "github_index":
-            # GitHub Pages index - use pip --find-links
+            # GitHub Pages index - try to resolve exact wheel URL first
             index_url = _substitute_template(config["index_url"], env)
             pkg_spec = f"{package}=={version}" if version else package
             log(f"  Installing {package} (github_index)...")
@@ -504,14 +507,17 @@ def _install_cuda_package(
             vars_dict = env.as_dict()
             wheel_url = resolve_wheel_from_index(index_url, package, vars_dict, version)
             if wheel_url:
+                # Install from resolved URL directly (guarantees we get what we resolved)
                 log(f"    Wheel: {wheel_url}")
+                _pip_install([wheel_url], no_deps=True, log=log)
             else:
+                # Fallback to find-links based resolution
                 log(f"    Find-links: {index_url}")
                 log(f"    Package: {pkg_spec}")
-            _pip_install_with_find_links(pkg_spec, index_url, log)
+                _pip_install_with_find_links(pkg_spec, index_url, log)
 
         elif method == "find_links":
-            # Generic find-links (e.g., PyG) - use pip --find-links
+            # Generic find-links (e.g., PyG) - try to resolve exact wheel URL first
             index_url = _substitute_template(config["index_url"], env)
             pkg_spec = f"{package}=={version}" if version else package
             log(f"  Installing {package} (find_links)...")
@@ -519,11 +525,14 @@ def _install_cuda_package(
             vars_dict = env.as_dict()
             wheel_url = resolve_wheel_from_index(index_url, package, vars_dict, version)
             if wheel_url:
+                # Install from resolved URL directly (guarantees we get what we resolved)
                 log(f"    Wheel: {wheel_url}")
+                _pip_install([wheel_url], no_deps=True, log=log)
             else:
+                # Fallback to find-links based resolution
                 log(f"    Find-links: {index_url}")
                 log(f"    Package: {pkg_spec}")
-            _pip_install_with_find_links(pkg_spec, index_url, log)
+                _pip_install_with_find_links(pkg_spec, index_url, log)
 
         elif method == "pypi_variant":
             # Transform package name based on CUDA version

comfy_env/pixi.py

@@ -276,6 +276,7 @@ def create_pixi_toml(
 
 def clean_pixi_artifacts(
     node_dir: Path,
+    env_name: Optional[str] = None,
     log: Callable[[str], None] = print,
 ) -> None:
     """
@@ -286,6 +287,7 @@ def clean_pixi_artifacts(
 
     Args:
         node_dir: Directory containing the pixi artifacts.
+        env_name: Environment name (for removing _env_ symlink).
         log: Logging callback.
     """
     pixi_toml = node_dir / "pixi.toml"
@@ -302,6 +304,13 @@ def clean_pixi_artifacts(
         shutil.rmtree(pixi_dir)
         log("  Removed previous .pixi/ directory")
 
+    # Remove _env_ symlink if it exists
+    if env_name:
+        symlink_path = node_dir / f"_env_{env_name}"
+        if symlink_path.is_symlink():
+            symlink_path.unlink()
+            log(f"  Removed previous _env_{env_name} symlink")
+
 
 def pixi_install(
     env_config: IsolatedEnv,
@@ -344,7 +353,7 @@ def pixi_install(
         return True
 
     # Clean previous pixi artifacts
-    clean_pixi_artifacts(node_dir, log)
+    clean_pixi_artifacts(node_dir, env_config.name, log)
 
     # Ensure pixi is installed
     pixi_path = ensure_pixi(log=log)
@@ -373,6 +382,22 @@ def pixi_install(
                 log(f"  {line}")
 
     log("pixi install completed successfully!")
+
+    # Create _env_{name} symlink for compatibility with uv backend
+    # This ensures code that expects _env_envname/bin/python works with pixi
+    symlink_path = node_dir / f"_env_{env_config.name}"
+    pixi_env_path = node_dir / ".pixi" / "envs" / "default"
+
+    if pixi_env_path.exists():
+        # Remove existing symlink or directory if present
+        if symlink_path.is_symlink():
+            symlink_path.unlink()
+        elif symlink_path.exists():
+            shutil.rmtree(symlink_path)
+
+        symlink_path.symlink_to(pixi_env_path)
+        log(f"Created symlink: _env_{env_config.name} -> .pixi/envs/default")
+
     return True
 
 

comfy_env/wheel_sources.yml

@@ -49,12 +49,12 @@ packages:
     description: Spline convolutions for PyTorch
 
   # ===========================================================================
-  # pytorch3d - Facebook's official wheels
-  # https://github.com/facebookresearch/pytorch3d/blob/main/INSTALL.md
+  # pytorch3d - MiroPsota's torch_packages_builder
+  # https://github.com/MiroPsota/torch_packages_builder
   # ===========================================================================
   pytorch3d:
     method: index
-    index_url: "https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py3{py_minor}_cu{cuda_short}_pyt{torch_short}/download.html"
+    index_url: "https://miropsota.github.io/torch_packages_builder/pytorch3d/"
     description: PyTorch3D - 3D deep learning library
 
   # ===========================================================================

{comfy_env-0.0.32.dist-info → comfy_env-0.0.34.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comfy-env
-Version: 0.0.32
+Version: 0.0.34
 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.32.dist-info → comfy_env-0.0.34.dist-info}/RECORD

@@ -3,16 +3,16 @@ comfy_env/cli.py,sha256=X-GCQMP0mtMcE3ZgkT-VLQ4Gq3UUvcb_Ux_NClEFhgI,15975
 comfy_env/decorator.py,sha256=6JCKwLHaZtOLVDexs_gh_-NtS2ZK0V7nGCPqkyeYEAA,16688
 comfy_env/errors.py,sha256=8hN8NDlo8oBUdapc-eT3ZluigI5VBzfqsSBvQdfWlz4,9943
 comfy_env/index_resolver.py,sha256=D8BttTJ7BOiukUvmdT6_dGdzFV3CahGL2m28X-HwPeE,4650
-comfy_env/install.py,sha256=zVrpo6bzWw2gSYTAzjp-6Ne75swLfOr-mhpROe6PhJY,29748
+comfy_env/install.py,sha256=eS3Q8T9bFP4AH2PfqLmjqWA8cu8vJsg02aXBTP5x1IQ,30446
 comfy_env/nodes.py,sha256=CWUe35jU5SKk4ur-SddZePdqWgxJDlxGhpcJiu5pAK4,4354
-comfy_env/pixi.py,sha256=_p3h9iFPHwNaEAEL8SeMJLaUmoyrroH4HJcbfG1Nh8A,13383
+comfy_env/pixi.py,sha256=y25mUDhB3bCqhPMGF0h23Tf8ZHykK4gLJrkvOhsPWmE,14398
 comfy_env/registry.py,sha256=uFCtGmWYvwGCqObXgzmArX7o5JsFNsHXxayofk3m6no,2569
 comfy_env/resolver.py,sha256=l-AnmCE1puG6CvdpDB-KrsfG_cn_3uO2DryYizUnG_4,12474
 comfy_env/env/__init__.py,sha256=imQdoQEQvrRT-QDtyNpFlkVbm2fBzgACdpQwRPd09fI,1157
 comfy_env/env/config.py,sha256=5rK7r2uRItMXJnKAn8DmVQoadLo2njHTuaxrWybhppU,7469
 comfy_env/env/config_file.py,sha256=1UdcL1TwKceGaSunCnsHiuPyxpCSq1JpelScUEsCBn8,23669
 comfy_env/env/cuda_gpu_detection.py,sha256=YLuXUdWg6FeKdNyLlQAHPlveg4rTenXJ2VbeAaEi9QE,9755
-comfy_env/env/manager.py,sha256=zBBN5P0rOrXzPodoF2294xsk4F1Gg7CWZsn2MkxlWkw,25678
+comfy_env/env/manager.py,sha256=wv_od7Rxa-qMlToUGzOhAsDvZY6C0P7y9ayqYiKKLPg,26460
 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
@@ -34,9 +34,9 @@ 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=4YSNPn7hALrvMVbkO4RkTeFTcc0lhfLMk5QTWjY4PHw,22134
 comfy_env/workers/venv.py,sha256=_ekHfZPqBIPY08DjqiXm6cTBQH4DrbxRWR3AAv3mit8,31589
-comfy_env/wheel_sources.yml,sha256=kZGK4-AJ0vvKmqDdxuiATLjGkaHHs-ansIckYION_nA,8871
-comfy_env-0.0.32.dist-info/METADATA,sha256=1IkItwJJn7EIvUCVfSTT3iq0MSh5d8PKsYxkk2LCtVo,5400
-comfy_env-0.0.32.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-comfy_env-0.0.32.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
-comfy_env-0.0.32.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
-comfy_env-0.0.32.dist-info/RECORD,,
+comfy_env/wheel_sources.yml,sha256=ep7husItiWik_8wUaVG5j1ttAw4NhA2daFjm2UCR7lA,8808
+comfy_env-0.0.34.dist-info/METADATA,sha256=gOWb3KyliLVyLwp12wZBZb3JkeG8MTxetq3oRJnDmIM,5400
+comfy_env-0.0.34.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+comfy_env-0.0.34.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
+comfy_env-0.0.34.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
+comfy_env-0.0.34.dist-info/RECORD,,