comfy-env 0.1.23__tar.gz → 0.1.25__tar.gz

{comfy_env-0.1.23 → comfy_env-0.1.25}/PKG-INFO

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

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

{comfy_env-0.1.23 → comfy_env-0.1.25}/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.23 → comfy_env-0.1.25}/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()

{comfy_env-0.1.23 → comfy_env-0.1.25}/src/comfy_env/packages/apt.py

@@ -32,29 +32,35 @@ def apt_install(packages: List[str], log: Callable[[str], None] = print) -> bool
     else:
         log("[apt] apt-get update succeeded")
 
-    # Install missing packages
-    log(f"[apt] Running: sudo apt-get install -y {' '.join(missing)}")
-    result = subprocess.run(
-        ["sudo", "apt-get", "install", "-y"] + missing,
-        capture_output=True, text=True
-    )
-
-    if result.returncode != 0:
-        log(f"[apt] ERROR: apt-get install failed (exit {result.returncode})")
-        log(f"[apt] stdout: {result.stdout}")
-        log(f"[apt] stderr: {result.stderr}")
-        return False
+    # Install packages ONE BY ONE so one failure doesn't block others
+    installed = []
+    failed = []
+    for pkg in missing:
+        log(f"[apt] Installing: {pkg}")
+        result = subprocess.run(
+            ["sudo", "apt-get", "install", "-y", pkg],
+            capture_output=True, text=True
+        )
+        if result.returncode != 0:
+            log(f"[apt] FAILED: {pkg} - {result.stderr.strip()}")
+            failed.append(pkg)
+        else:
+            log(f"[apt] OK: {pkg}")
+            installed.append(pkg)
 
-    log("[apt] Installation succeeded")
+    # Summary
+    if installed:
+        log(f"[apt] Successfully installed: {installed}")
+    if failed:
+        log(f"[apt] Failed to install: {failed}")
 
-    # Verify installation
-    still_missing = check_apt_packages(missing)
+    # Verify what's actually installed now
+    still_missing = check_apt_packages(packages)
     if still_missing:
-        log(f"[apt] WARNING: These packages still not found: {still_missing}")
-        return False
+        log(f"[apt] WARNING: These packages are not available: {still_missing}")
 
-    log("[apt] All packages verified installed")
-    return True
+    # Return True if we installed at least something (partial success is OK)
+    return len(installed) > 0 or len(missing) == len(failed)
 
 
 def check_apt_packages(packages: List[str]) -> List[str]: