comfy-env 0.1.17__tar.gz → 0.1.18__tar.gz

{comfy_env-0.1.17 → comfy_env-0.1.18}/PKG-INFO

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

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

{comfy_env-0.1.17 → comfy_env-0.1.18}/src/comfy_env/isolation/workers/subprocess.py

@@ -193,6 +193,32 @@ from multiprocessing import shared_memory as shm
 import numpy as np
 
 
+def _prepare_trimesh_for_pickle(mesh):
+    """
+    Prepare a trimesh object for cross-Python-version pickling.
+
+    Trimesh attaches helper objects (ray tracer, proximity query) that may use
+    native extensions like embreex. These cause import errors when unpickling
+    on a system without those extensions. We strip them - they'll be recreated
+    lazily when needed.
+
+    Note: Do NOT strip _cache - trimesh needs it to function properly.
+    """
+    # Make a copy to avoid modifying the original
+    mesh = mesh.copy()
+
+    # Remove helper objects that may have unpickleable native code references
+    # These are lazily recreated on first access anyway
+    # Do NOT remove _cache - it's needed for trimesh to work
+    for attr in ('ray', '_ray', 'permutate', 'nearest'):
+        try:
+            delattr(mesh, attr)
+        except AttributeError:
+            pass
+
+    return mesh
+
+
 def _to_shm(obj, registry, visited=None):
     """
     Serialize object to shared memory. Returns JSON-safe metadata.
@@ -231,6 +257,7 @@ def _to_shm(obj, registry, visited=None):
     # trimesh.Trimesh -> pickle -> shared memory (preserves visual, metadata, normals)
     if t == 'Trimesh':
         import pickle
+        obj = _prepare_trimesh_for_pickle(obj)
         mesh_bytes = pickle.dumps(obj)
 
         block = shm.SharedMemory(create=True, size=len(mesh_bytes))
@@ -533,6 +560,7 @@ def _to_shm(obj, registry, visited=None):
     # trimesh.Trimesh -> pickle -> shared memory (preserves visual, metadata, normals)
     if t == 'Trimesh':
         import pickle
+        obj = _prepare_trimesh_for_pickle(obj)
         mesh_bytes = pickle.dumps(obj)
 
         block = shm.SharedMemory(create=True, size=len(mesh_bytes))
@@ -642,31 +670,6 @@ def _should_use_reference(obj):
     # Everything else (custom classes) - pass by reference
     return True
 
-def _prepare_trimesh_for_pickle(mesh):
-    """
-    Prepare a trimesh object for cross-Python-version pickling.
-
-    Trimesh attaches helper objects (ray tracer, proximity query) that may use
-    native extensions like embreex. These cause import errors when unpickling
-    on a system without those extensions. We strip them - they'll be recreated
-    lazily when needed.
-
-    Note: Do NOT strip _cache - trimesh needs it to function properly.
-    """
-    # Make a copy to avoid modifying the original
-    mesh = mesh.copy()
-
-    # Remove helper objects that may have unpickleable native code references
-    # These are lazily recreated on first access anyway
-    # Do NOT remove _cache - it's needed for trimesh to work
-    for attr in ('ray', '_ray', 'permutate', 'nearest'):
-        try:
-            delattr(mesh, attr)
-        except AttributeError:
-            pass
-
-    return mesh
-
 
 def _serialize_result(obj, visited=None):
     """Convert result for IPC - complex objects become references."""

{comfy_env-0.1.17 → comfy_env-0.1.18}/src/comfy_env/packages/__init__.py

@@ -5,7 +5,6 @@ Handles pixi, CUDA wheels, apt packages, and node dependencies.
 """
 
 from .pixi import (
-    PIXI_VERSION,
     ensure_pixi,
     get_pixi_path,
     get_pixi_python,
@@ -35,7 +34,6 @@ from .node_dependencies import (
 
 __all__ = [
     # Pixi package manager
-    "PIXI_VERSION",
     "ensure_pixi",
     "get_pixi_path",
     "get_pixi_python",