comfy-env 0.1.24__tar.gz → 0.1.26__tar.gz

{comfy_env-0.1.24 → comfy_env-0.1.26}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: comfy-env
-Version: 0.1.24
+Version: 0.1.26
 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
@@ -19,6 +19,7 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.10
 Requires-Dist: numpy
 Requires-Dist: pip>=21.0
+Requires-Dist: pixi>=0.40.0
 Requires-Dist: tomli-w>=1.0.0
 Requires-Dist: tomli>=2.0.0
 Requires-Dist: uv>=0.4.0

{comfy_env-0.1.24 → comfy_env-0.1.26}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "comfy-env"
-version = "0.1.24"
+version = "0.1.26"
 description = "Environment management for ComfyUI custom nodes - CUDA wheel resolution and process isolation"
 readme = "README.md"
 license = {text = "MIT"}
@@ -22,6 +22,7 @@ dependencies = [
     "tomli>=2.0.0",
     "tomli-w>=1.0.0",
     "uv>=0.4.0",
+    "pixi>=0.40.0",
     "pip>=21.0",
     "numpy",
 ]

{comfy_env-0.1.24 → comfy_env-0.1.26}/src/comfy_env/environment/cache.py

@@ -2,6 +2,7 @@
 
 import glob
 import hashlib
+import os
 import shutil
 import sys
 from datetime import datetime
@@ -16,14 +17,16 @@ try:
 except ImportError:
     __version__ = "0.0.0-dev"
 
-CACHE_DIR = Path.home() / ".comfy-env" / "envs"
+CACHE_DIR = Path.home() / ".comfy-env" / "envs"  # Default, use get_cache_dir() for dynamic lookup
 MARKER_FILE = ".comfy-env-marker.toml"
 METADATA_FILE = ".comfy-env-metadata.toml"
 
 
 def get_cache_dir() -> Path:
-    CACHE_DIR.mkdir(parents=True, exist_ok=True)
-    return CACHE_DIR
+    """Get cache dir, checking COMFY_ENV_CACHE_DIR env var each time."""
+    cache_dir = Path(os.environ.get("COMFY_ENV_CACHE_DIR", Path.home() / ".comfy-env" / "envs"))
+    cache_dir.mkdir(parents=True, exist_ok=True)
+    return cache_dir
 
 
 def compute_config_hash(config_path: Path) -> str:

{comfy_env-0.1.24 → comfy_env-0.1.26}/src/comfy_env/install.py

@@ -151,7 +151,7 @@ def _install_via_pixi(cfg: ComfyEnvConfig, node_dir: Path, log: Callable[[str],
     log("Running pixi install...")
     result = subprocess.run([str(pixi_path), "install"], cwd=node_dir, capture_output=True, text=True)
     if result.returncode != 0:
-        raise RuntimeError(f"pixi install failed: {result.stderr}")
+        raise RuntimeError(f"pixi install failed:\nstderr: {result.stderr}\nstdout: {result.stdout}")
 
     if cfg.cuda_packages and cuda_version:
         log(f"Installing CUDA packages...")
@@ -171,7 +171,7 @@ def _install_via_pixi(cfg: ComfyEnvConfig, node_dir: Path, log: Callable[[str],
             result = subprocess.run([str(python_path), "-m", "pip", "install", "--no-deps", "--no-cache-dir", wheel_url],
                                    capture_output=True, text=True)
             if result.returncode != 0:
-                raise RuntimeError(f"Failed: {result.stderr}")
+                raise RuntimeError(f"Failed to install {package}:\nstderr: {result.stderr}\nstdout: {result.stdout}")
 
     # Find config file for marker
     config_path = node_dir / CONFIG_FILE_NAME

{comfy_env-0.1.24 → comfy_env-0.1.26}/src/comfy_env/isolation/workers/subprocess.py

@@ -668,7 +668,10 @@ def _to_shm(obj, registry, visited=None):
         visited = {}
     obj_id = id(obj)
     if obj_id in visited:
-        return visited[obj_id]
+        cached = visited[obj_id]
+        if isinstance(cached, dict) and cached.get("__type__") == "TensorRef":
+            print(f"[SHM DEBUG] _to_shm CACHE HIT: id={obj_id} -> storage_key=...{cached.get('storage_key','?')[-20:]}, tensor_size={cached.get('tensor_size')}", file=sys.stderr)
+        return cached
     t = type(obj).__name__
 
     # Tensor -> use PyTorch's native shared memory (bypasses resource_tracker)
@@ -676,6 +679,7 @@ def _to_shm(obj, registry, visited=None):
         import torch
         tensor = obj.detach().cpu().contiguous()
         result = _serialize_tensor_native(tensor, registry)
+        print(f"[SHM DEBUG] _to_shm Tensor: id={obj_id}, orig_shape={list(obj.shape)}, new_shape={list(tensor.shape)} -> storage_key=...{result.get('storage_key','?')[-20:]}, tensor_size={result.get('tensor_size')}", file=sys.stderr)
         visited[obj_id] = result
         return result
 
@@ -769,6 +773,7 @@ def _from_shm(obj):
     # TensorRef -> use PyTorch's native deserialization (new format, worker->parent)
     if obj.get("__type__") == "TensorRef":
         tensor = _deserialize_tensor_native(obj)
+        print(f"[SHM DEBUG] _from_shm TensorRef: storage_key=...{obj.get('storage_key','?')[-20:]}, expected_size={obj.get('tensor_size')} -> actual_shape={list(tensor.shape)}", file=sys.stderr)
         # Convert back to numpy if it was originally numpy
         if obj.get("__was_numpy__"):
             return tensor.numpy()