comfy-env 0.1.17__py3-none-any.whl → 0.1.19__py3-none-any.whl

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))
@@ -506,6 +533,21 @@ if sys.platform == "win32":
 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.
+    Strips native extension helpers that cause import errors.
+    """
+    mesh = mesh.copy()
+    for attr in ('ray', '_ray', 'permutate', 'nearest'):
+        try:
+            delattr(mesh, attr)
+        except AttributeError:
+            pass
+    return mesh
+
+
 def _to_shm(obj, registry, visited=None):
     """Serialize to shared memory. Returns JSON-safe metadata."""
     if visited is None:
@@ -533,6 +575,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 +685,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/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",

{comfy_env-0.1.17.dist-info → comfy_env-0.1.19.dist-info}/METADATA

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

@@ -20,8 +20,8 @@ comfy_env/isolation/wrap.py,sha256=K7GAkqU_Uxe717eUtPsFv5kcr_Jfbh3x79A-8vbY1nY,8
 comfy_env/isolation/workers/__init__.py,sha256=Zp6sZSRBcb5Negqgzqs3jPjfO9T1u3nNrQhp6WqTAuc,325
 comfy_env/isolation/workers/base.py,sha256=4ZYTaQ4J0kBHCoO_OfZnsowm4rJCoqinZUaOtgkOPbw,2307
 comfy_env/isolation/workers/mp.py,sha256=ygOgx2iyLN7l5fWkKI4lqzQsDyfAAd9Gb4gTYLp7o1A,34061
-comfy_env/isolation/workers/subprocess.py,sha256=ML6I9IenReagP8iT0Cd2ipet6JPK1gnDbOianOuFwOw,57164
-comfy_env/packages/__init__.py,sha256=6PTwUfUdJDTbIw46dCiA42qk4zUe_gw29xOaklBiMMc,1193
+comfy_env/isolation/workers/subprocess.py,sha256=15G4rQA7bO-mmVkkeeR3we9ytHIw6kHLNdjc9XWXCu8,57650
+comfy_env/packages/__init__.py,sha256=4pRCUnfcVFVgy7hkbPz9BPVXELtSFHha6L7n-hqNuZA,1155
 comfy_env/packages/apt.py,sha256=pxy3A5ZHv3X8ExCVyohODY8Fcy9ji4izIVPfYoxhqT4,1027
 comfy_env/packages/cuda_wheels.py,sha256=G_CnlwNcfeWlEU24aCVBpeqQQ05y8_02dDLBwBFNwII,3980
 comfy_env/packages/node_dependencies.py,sha256=AX_CY6j43tTY5KhyPfU7Wz6zgLAfWF0o0JkTrcNSecg,2966
@@ -29,8 +29,8 @@ comfy_env/packages/pixi.py,sha256=RPu8x5sSOLE1CYAhWMMjoQrbFGGt00fdsbqtRcTz7LQ,38
 comfy_env/packages/toml_generator.py,sha256=Vhc8F9euHhMTwH1TV6t96-D9Pjrn9jIN4e9WXrCIFE8,3414
 comfy_env/templates/comfy-env-instructions.txt,sha256=ve1RAthW7ouumU9h6DM7mIRX1MS8_Tyonq2U4tcrFu8,1031
 comfy_env/templates/comfy-env.toml,sha256=ROIqi4BlPL1MEdL1VgebfTHpdwPNYGHwWeigI9Kw-1I,4831
-comfy_env-0.1.17.dist-info/METADATA,sha256=XlM9mAERf1U2q9sBbrmY7jIgqrl8IsLrs8vK9CmLnpw,4808
-comfy_env-0.1.17.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-comfy_env-0.1.17.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
-comfy_env-0.1.17.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
-comfy_env-0.1.17.dist-info/RECORD,,
+comfy_env-0.1.19.dist-info/METADATA,sha256=H3iq92VDNsR5lP_EndbQq9S0ldsB2-WUzNb8PBByE7I,4808
+comfy_env-0.1.19.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+comfy_env-0.1.19.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
+comfy_env-0.1.19.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
+comfy_env-0.1.19.dist-info/RECORD,,