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

comfy_env/install.py

@@ -484,19 +484,31 @@ def _install_cuda_package(
             else:
                 pkg_spec = f"{package}=={version}" if version else package
             log(f"  Installing {package} (index)...")
-            # Try to resolve exact wheel URL from index
-            actual_version = resolved_version if "version_template" in config else version
-            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)
+
+            # Resolve version: use provided version, default_version, or None
+            effective_version = version if version and version != "*" else config.get("default_version")
+
+            # Check for wheel_template first (direct URL construction, no index parsing)
+            if "wheel_template" in config and effective_version:
+                vars_dict = env.as_dict()
+                vars_dict["version"] = effective_version
+                wheel_url = _substitute_template(config["wheel_template"], vars_dict)
                 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)
+                # Try to resolve exact wheel URL from index
+                actual_version = resolved_version if "version_template" in config else version
+                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:
+                    raise InstallError(
+                        f"Failed to resolve wheel URL for {package} from index {index_url}. "
+                        "No matching wheel found and PyPI fallback is disabled.",
+                    )
 
         elif method == "github_index":
             # GitHub Pages index - try to resolve exact wheel URL first

comfy_env/wheel_sources.yml

@@ -55,6 +55,8 @@ packages:
   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"
+    default_version: "0.7.9"
     description: PyTorch3D - 3D deep learning library
 
   # ===========================================================================

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

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

@@ -3,7 +3,7 @@ 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=eS3Q8T9bFP4AH2PfqLmjqWA8cu8vJsg02aXBTP5x1IQ,30446
+comfy_env/install.py,sha256=SeYcnvFErBpAvTMG1L02PMKt39yrD_9WHGiFfaJ_igA,31172
 comfy_env/nodes.py,sha256=CWUe35jU5SKk4ur-SddZePdqWgxJDlxGhpcJiu5pAK4,4354
 comfy_env/pixi.py,sha256=y25mUDhB3bCqhPMGF0h23Tf8ZHykK4gLJrkvOhsPWmE,14398
 comfy_env/registry.py,sha256=uFCtGmWYvwGCqObXgzmArX7o5JsFNsHXxayofk3m6no,2569
@@ -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=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,,
+comfy_env/wheel_sources.yml,sha256=WHSSgbEXDKgRdJAi5FKiEU0l5dFxP_1p4JThc1bbqj8,9007
+comfy_env-0.0.36.dist-info/METADATA,sha256=5Eyt-3TBXtZ_G4eVdCMe7FehkcTo7EPcSnn-Zfz8oqg,5400
+comfy_env-0.0.36.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+comfy_env-0.0.36.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
+comfy_env-0.0.36.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
+comfy_env-0.0.36.dist-info/RECORD,,