comfy-env 0.0.51__py3-none-any.whl → 0.0.52__py3-none-any.whl

comfy_env/pixi.py

@@ -246,6 +246,10 @@ def create_pixi_toml(
     lines.append("[dependencies]")
     lines.append(f'python = "{env_config.python}.*"')
 
+    # On Windows, use MKL BLAS to avoid OpenBLAS crashes (numpy blas_fpe_check issue)
+    if sys.platform == "win32":
+        lines.append('libblas = { version = "*", build = "*mkl" }')
+
     for pkg in conda.packages:
         # Parse package spec (name=version or name>=version or just name)
         if "=" in pkg and not pkg.startswith("="):

comfy_env/stub_imports.py

@@ -156,19 +156,26 @@ def _get_import_names_from_pixi(node_dir: Path) -> Set[str]:
     """
     import_names = set()
 
-    # Find the pixi site-packages
-    pixi_lib = node_dir / ".pixi" / "envs" / "default" / "lib"
+    pixi_base = node_dir / ".pixi" / "envs" / "default"
 
-    if not pixi_lib.exists():
-        return import_names
-
-    # Find the python version directory (e.g., python3.11)
-    python_dirs = list(pixi_lib.glob("python3.*"))
-    if not python_dirs:
-        return import_names
+    # Find site-packages (different paths on Windows vs Linux)
+    # Linux: .pixi/envs/default/lib/python3.x/site-packages
+    # Windows: .pixi/envs/default/Lib/site-packages
+    site_packages = None
 
-    site_packages = python_dirs[0] / "site-packages"
-    if not site_packages.exists():
+    # Try Windows path first (Lib/site-packages)
+    win_site = pixi_base / "Lib" / "site-packages"
+    if win_site.exists():
+        site_packages = win_site
+    else:
+        # Try Linux path (lib/python3.x/site-packages)
+        pixi_lib = pixi_base / "lib"
+        if pixi_lib.exists():
+            python_dirs = list(pixi_lib.glob("python3.*"))
+            if python_dirs:
+                site_packages = python_dirs[0] / "site-packages"
+
+    if site_packages is None or not site_packages.exists():
         return import_names
 
     # Scan for importable modules

comfy_env/workers/venv.py

@@ -252,10 +252,29 @@ def _serialize_for_ipc(obj, visited=None):
 # Worker script template - minimal, runs in target venv
 _WORKER_SCRIPT = '''
 import sys
+import os
 import json
 import traceback
 from types import SimpleNamespace
 
+# On Windows, add DLL directories for proper library loading
+if sys.platform == "win32" and hasattr(os, "add_dll_directory"):
+    _host_python_dir = os.environ.get("COMFYUI_HOST_PYTHON_DIR")
+    if _host_python_dir:
+        try:
+            os.add_dll_directory(_host_python_dir)
+            _dlls_dir = os.path.join(_host_python_dir, "DLLs")
+            if os.path.isdir(_dlls_dir):
+                os.add_dll_directory(_dlls_dir)
+        except Exception:
+            pass
+    _pixi_library_bin = os.environ.get("COMFYUI_PIXI_LIBRARY_BIN")
+    if _pixi_library_bin:
+        try:
+            os.add_dll_directory(_pixi_library_bin)
+        except Exception:
+            pass
+
 def _deserialize_isolated_objects(obj):
     """Reconstruct objects serialized with __isolated_object__ marker."""
     if isinstance(obj, dict):
@@ -475,12 +494,29 @@ class VenvWorker(Worker):
             env.update(self.extra_env)
             env["COMFYUI_ISOLATION_WORKER"] = "1"
 
-            # For conda/pixi environments, add lib dir to LD_LIBRARY_PATH
+            # For conda/pixi environments, add lib dir to LD_LIBRARY_PATH (Linux)
             lib_dir = self.python.parent.parent / "lib"
             if lib_dir.is_dir():
                 existing = env.get("LD_LIBRARY_PATH", "")
                 env["LD_LIBRARY_PATH"] = f"{lib_dir}:{existing}" if existing else str(lib_dir)
 
+            # On Windows, pass host Python directory and pixi Library/bin for DLL loading
+            if sys.platform == "win32":
+                env["COMFYUI_HOST_PYTHON_DIR"] = str(Path(sys.executable).parent)
+
+                # For pixi environments with MKL, add Library/bin to PATH for DLL loading
+                # Pixi has python.exe directly in env dir, not in Scripts/
+                env_dir = self.python.parent
+                library_bin = env_dir / "Library" / "bin"
+                if library_bin.is_dir():
+                    existing_path = env.get("PATH", "")
+                    env["PATH"] = f"{env_dir};{library_bin};{existing_path}"
+                    env["COMFYUI_PIXI_LIBRARY_BIN"] = str(library_bin)
+                    # Allow duplicate OpenMP libraries (MKL's libiomp5md.dll + PyTorch's libomp.dll)
+                    env["KMP_DUPLICATE_LIB_OK"] = "TRUE"
+                    # Use UTF-8 encoding for stdout/stderr to handle Unicode symbols
+                    env["PYTHONIOENCODING"] = "utf-8"
+
             # Run subprocess
             cmd = [
                 str(self.python),
@@ -592,6 +628,15 @@ if sys.platform == "win32":
         except Exception:
             pass
 
+    # For pixi environments with MKL, add Library/bin for MKL DLLs
+    _pixi_library_bin = os.environ.get("COMFYUI_PIXI_LIBRARY_BIN")
+    if _pixi_library_bin and hasattr(os, "add_dll_directory"):
+        try:
+            os.add_dll_directory(_pixi_library_bin)
+            print(f"[worker] Added pixi Library/bin to DLL search: {_pixi_library_bin}", flush=True)
+        except Exception as e:
+            print(f"[worker] Failed to add pixi Library/bin: {e}", flush=True)
+
 # =============================================================================
 # Object Reference System - keep complex objects in worker, pass refs to host
 # =============================================================================
@@ -977,6 +1022,22 @@ class PersistentVenvWorker(Worker):
         if sys.platform == "win32":
             env["COMFYUI_HOST_PYTHON_DIR"] = str(Path(sys.executable).parent)
 
+            # For pixi environments with MKL, add Library/bin to PATH for DLL loading
+            # MKL DLLs are in .pixi/envs/default/Library/bin/
+            # Pixi has python.exe directly in env dir, not in Scripts/
+            env_dir = self.python.parent
+            library_bin = env_dir / "Library" / "bin"
+            if library_bin.is_dir():
+                existing_path = env.get("PATH", "")
+                # Add env dir and Library/bin to PATH
+                env["PATH"] = f"{env_dir};{library_bin};{existing_path}"
+                # Also pass as env var so worker can use os.add_dll_directory()
+                env["COMFYUI_PIXI_LIBRARY_BIN"] = str(library_bin)
+                # Allow duplicate OpenMP libraries (MKL's libiomp5md.dll + PyTorch's libomp.dll)
+                env["KMP_DUPLICATE_LIB_OK"] = "TRUE"
+                # Use UTF-8 encoding for stdout/stderr to handle Unicode symbols
+                env["PYTHONIOENCODING"] = "utf-8"
+
         # Find ComfyUI base and set env var for folder_paths stub
         comfyui_base = self._find_comfyui_base()
         if comfyui_base:

{comfy_env-0.0.51.dist-info → comfy_env-0.0.52.dist-info}/METADATA

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

@@ -5,10 +5,10 @@ comfy_env/errors.py,sha256=8hN8NDlo8oBUdapc-eT3ZluigI5VBzfqsSBvQdfWlz4,9943
 comfy_env/install.py,sha256=b2eXFqQ58pUBWpwc1AhymbTh7dGRMnEuq5FoZot0CO0,15661
 comfy_env/isolation.py,sha256=wuze8TmkRbdMfrg2cj9quoGYJWsxKcc5iiDvmLWX16g,8680
 comfy_env/nodes.py,sha256=CWUe35jU5SKk4ur-SddZePdqWgxJDlxGhpcJiu5pAK4,4354
-comfy_env/pixi.py,sha256=p3qP-4l95-toOI3-N2luJQ7kB-KLbalDB6yEJjmfzoc,19574
+comfy_env/pixi.py,sha256=qycJVCwtxxG5a9w3TGiRX3sw_GcEsIiDEy9hu_H7loU,19765
 comfy_env/registry.py,sha256=w-QwvAPFlCrBYRAv4cXkp2zujQPZn8Fk5DUxKCtox8o,3430
 comfy_env/resolver.py,sha256=WoNIo2IfTR2RlEf_HQl66eAeMa2R2pmLof_UdK-0RNE,6714
-comfy_env/stub_imports.py,sha256=s84q8x5156ZkHNJhbuDqMhU_c-0ux51Rhe1aoVcGrj4,8920
+comfy_env/stub_imports.py,sha256=9aSWkgDHU0GjqQ8BmUMJyKewc47OtlbWPbAYY0gkCrM,9274
 comfy_env/env/__init__.py,sha256=imQdoQEQvrRT-QDtyNpFlkVbm2fBzgACdpQwRPd09fI,1157
 comfy_env/env/config.py,sha256=cRSuWiVZGSK9xpvyNIyMirlHyfTm6_AB_eXzeKyvfnc,7749
 comfy_env/env/config_file.py,sha256=g6fZq6XbTUXx97meWW_7xfKcuC0sN002c7fBvR0eCOg,24819
@@ -39,10 +39,10 @@ comfy_env/workers/base.py,sha256=ZILYXlvGCWuCZXmjKqfG8VeD19ihdYaASdlbasl2BMo,231
 comfy_env/workers/pool.py,sha256=MtjeOWfvHSCockq8j1gfnxIl-t01GSB79T5N4YB82Lg,6956
 comfy_env/workers/tensor_utils.py,sha256=TCuOAjJymrSbkgfyvcKtQ_KbVWTqSwP9VH_bCaFLLq8,6409
 comfy_env/workers/torch_mp.py,sha256=TnsCoBHEJBXEoBkx7WiCd9tBAlzFtMOw1dk_7_zGJZY,22288
-comfy_env/workers/venv.py,sha256=MXN0KRlIPREEgJKM-4_sXHQw4Dv8hLFGohmway6y3UU,46393
+comfy_env/workers/venv.py,sha256=IFeMaOyxTW3RXfc71Q1i_JhfE7ZBbTGm18Rr_KLI1cE,49606
 comfy_env/wheel_sources.yml,sha256=uU0YJmWaiLAicQNN9VYS8PZevlP2NOH6mBUE294dvAo,8156
-comfy_env-0.0.51.dist-info/METADATA,sha256=6DlmPszS1Q0lDXK1I0YXfkbaKPQyZKiZN7PsXQuVSHk,8735
-comfy_env-0.0.51.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-comfy_env-0.0.51.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
-comfy_env-0.0.51.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
-comfy_env-0.0.51.dist-info/RECORD,,
+comfy_env-0.0.52.dist-info/METADATA,sha256=afUGkDhRWskd-PWfDjXCHUUCqoMJNE9933-EV7JyDOg,8735
+comfy_env-0.0.52.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+comfy_env-0.0.52.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
+comfy_env-0.0.52.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
+comfy_env-0.0.52.dist-info/RECORD,,