comfy-env 0.0.36__tar.gz → 0.0.37__tar.gz

{comfy_env-0.0.36 → comfy_env-0.0.37}/PKG-INFO

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

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

{comfy_env-0.0.36 → comfy_env-0.0.37}/src/comfy_env/stubs/folder_paths.py

@@ -11,7 +11,7 @@ from pathlib import Path
 _comfyui_base = None
 
 def _find_comfyui_base():
-    """Find ComfyUI base from COMFYUI_BASE env var or by walking up."""
+    """Find ComfyUI base from COMFYUI_BASE env var, child dirs, or by walking up."""
     global _comfyui_base
     if _comfyui_base:
         return _comfyui_base
@@ -21,8 +21,18 @@ def _find_comfyui_base():
         _comfyui_base = Path(os.environ["COMFYUI_BASE"])
         return _comfyui_base
 
+    # Check common child directories (for test environments)
+    # Also check parent's children (isolated venv is sibling to .comfy-test-env)
+    cwd = Path.cwd().resolve()
+    for base in [cwd, cwd.parent]:
+        for child in [".comfy-test-env/ComfyUI", "ComfyUI"]:
+            candidate = base / child
+            if (candidate / "main.py").exists() and (candidate / "comfy").exists():
+                _comfyui_base = candidate
+                return _comfyui_base
+
     # Walk up from cwd looking for ComfyUI
-    current = Path.cwd().resolve()
+    current = cwd
     for _ in range(10):
         if (current / "main.py").exists() and (current / "comfy").exists():
             _comfyui_base = current

{comfy_env-0.0.36 → comfy_env-0.0.37}/src/comfy_env/workers/venv.py

@@ -594,7 +594,16 @@ class PersistentVenvWorker(Worker):
         self._worker_script.write_text(_PERSISTENT_WORKER_SCRIPT)
 
     def _find_comfyui_base(self) -> Optional[Path]:
-        """Find ComfyUI base directory by walking up from working_dir."""
+        """Find ComfyUI base directory."""
+        # Check common child directories (for test environments)
+        # Also check parent's children (isolated venv is sibling to .comfy-test-env)
+        for base in [self.working_dir, self.working_dir.parent]:
+            for child in [".comfy-test-env/ComfyUI", "ComfyUI"]:
+                candidate = base / child
+                if (candidate / "main.py").exists() and (candidate / "comfy").exists():
+                    return candidate
+
+        # Walk up from working_dir (standard ComfyUI custom_nodes layout)
         current = self.working_dir.resolve()
         for _ in range(10):
             if (current / "main.py").exists() and (current / "comfy").exists():