comfy-env 0.0.57__tar.gz → 0.0.59__tar.gz

{comfy_env-0.0.57 → comfy_env-0.0.59}/PKG-INFO

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

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

{comfy_env-0.0.57 → comfy_env-0.0.59}/src/comfy_env/env/config.py

@@ -156,6 +156,8 @@ class IsolatedEnv:
     conda: Optional["CondaConfig"] = None
     # Runtime isolation - run node FUNCTION methods in isolated subprocess
     isolated: bool = False
+    # Environment variables for install (e.g., CMAKE_ARGS for build flags)
+    env_vars: Dict[str, str] = field(default_factory=dict)
 
     def __post_init__(self):
         """Validate and normalize configuration."""

{comfy_env-0.0.57 → comfy_env-0.0.59}/src/comfy_env/env/config_file.py

@@ -575,6 +575,13 @@ def _parse_single_env(name: str, env_data: Dict[str, Any], base_dir: Path) -> Is
     # Parse isolated flag for runtime process isolation
     isolated = env_data.get("isolated", False)
 
+    # Parse environment variables section [envname.env]
+    env_vars = {}
+    env_section = env_data.get("env", {})
+    if isinstance(env_section, dict):
+        # Filter to only string/numeric values (avoid nested tables like cuda/packages)
+        env_vars = {k: str(v) for k, v in env_section.items() if isinstance(v, (str, int, float))}
+
     return IsolatedEnv(
         name=name,
         python=python,
@@ -587,6 +594,7 @@ def _parse_single_env(name: str, env_data: Dict[str, Any], base_dir: Path) -> Is
         darwin_requirements=darwin_reqs,
         conda=conda_config,
         isolated=isolated,
+        env_vars=env_vars,
     )
 
 

{comfy_env-0.0.57 → comfy_env-0.0.59}/src/comfy_env/pixi.py

@@ -320,13 +320,13 @@ def create_pixi_toml(
     lines.append(f"channels = [{channels_str}]")
 
     # Platforms
+    # Note: On macOS we always use osx-64 (x86_64) even on ARM64 Macs.
+    # This runs under Rosetta 2 but ensures compatibility with packages
+    # that only have x86_64 wheels (e.g., embreex for trimesh raytracing).
     if sys.platform == "linux":
         lines.append('platforms = ["linux-64"]')
     elif sys.platform == "darwin":
-        if platform.machine() == "arm64":
-            lines.append('platforms = ["osx-arm64"]')
-        else:
-            lines.append('platforms = ["osx-64"]')
+        lines.append('platforms = ["osx-64"]')
     elif sys.platform == "win32":
         lines.append('platforms = ["win-64"]')
 
@@ -342,6 +342,7 @@ def create_pixi_toml(
     # Dependencies section (conda packages)
     lines.append("[dependencies]")
     lines.append(f'python = "{env_config.python}.*"')
+    lines.append('pip = "*"')  # Required for installing CUDA packages with --no-deps
 
     # On Windows, use MKL BLAS to avoid OpenBLAS crashes (numpy blas_fpe_check issue)
     if sys.platform == "win32":
@@ -574,11 +575,19 @@ def pixi_install(
 
     # Run pixi install
     log("Running pixi install...")
+
+    # Build environment with custom vars from config
+    install_env = os.environ.copy()
+    if env_config.env_vars:
+        install_env.update(env_config.env_vars)
+        log(f"  Using custom env vars: {list(env_config.env_vars.keys())}")
+
     result = subprocess.run(
         [str(pixi_path), "install"],
         cwd=node_dir,
         capture_output=True,
         text=True,
+        env=install_env,
     )
 
     if result.returncode != 0: