comfy-env 0.0.52__tar.gz → 0.0.53__tar.gz

{comfy_env-0.0.52 → comfy_env-0.0.53}/PKG-INFO

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

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

{comfy_env-0.0.52 → comfy_env-0.0.53}/src/comfy_env/pixi.py

@@ -203,10 +203,12 @@ def create_pixi_toml(
     Returns:
         Path to the generated pixi.toml file.
     """
-    if not env_config.conda:
-        raise ValueError("Environment has no conda configuration")
-
-    conda = env_config.conda
+    # Conda is optional - use defaults if not present
+    if env_config.conda:
+        conda = env_config.conda
+    else:
+        from comfy_env.env.config import CondaConfig
+        conda = CondaConfig(channels=["conda-forge"], packages=[])
     pixi_toml_path = node_dir / "pixi.toml"
 
     # Build pixi.toml content
@@ -375,6 +377,34 @@ def create_pixi_toml(
             lines.append(f'{name} = {value}')
 
         for dep in pypi_deps:
+            # Handle git dependencies in two formats:
+            # 1. pkg @ git+https://github.com/user/repo.git@commit
+            # 2. git+https://github.com/user/repo.git@commit (extract name from URL)
+            if "git+" in dep:
+                if " @ git+" in dep:
+                    # Format: pkg @ git+URL@commit
+                    match = re.match(r'^([a-zA-Z0-9._-]+)\s*@\s*git\+(.+?)(?:@([a-f0-9]+))?$', dep)
+                    if match:
+                        pkg_name = match.group(1)
+                        git_url = match.group(2)
+                        rev = match.group(3)
+                else:
+                    # Format: git+URL@commit (extract package name from repo name)
+                    match = re.match(r'^git\+(.+?)(?:@([a-f0-9]+))?$', dep)
+                    if match:
+                        git_url = match.group(1)
+                        rev = match.group(2)
+                        # Extract package name from URL (repo name without .git)
+                        repo_match = re.search(r'/([^/]+?)(?:\.git)?$', git_url)
+                        pkg_name = repo_match.group(1) if repo_match else git_url.split('/')[-1].replace('.git', '')
+
+                if match:
+                    if rev:
+                        lines.append(f'{pkg_name} = {{ git = "{git_url}", rev = "{rev}" }}')
+                    else:
+                        lines.append(f'{pkg_name} = {{ git = "{git_url}" }}')
+                    continue
+
             # Parse pip requirement format to pixi format
             # Handles extras like trimesh[easy]>=4.0.0
             name, version_spec, extras = _parse_pypi_requirement(dep)

{comfy_env-0.0.52 → comfy_env-0.0.53}/src/comfy_env/workers/venv.py

@@ -671,8 +671,12 @@ def _should_use_reference(obj):
     # Primitives - pass by value
     if isinstance(obj, (bool, int, float, str, bytes)):
         return False
-    # NumPy arrays and torch tensors - pass by value (they serialize well)
+    # NumPy scalars - pass by value (convert to Python primitives)
     obj_type = type(obj).__name__
+    if obj_type in ('float16', 'float32', 'float64', 'int8', 'int16', 'int32', 'int64',
+                    'uint8', 'uint16', 'uint32', 'uint64', 'bool_'):
+        return False
+    # NumPy arrays and torch tensors - pass by value (they serialize well)
     if obj_type in ('ndarray', 'Tensor'):
         return False
     # Dicts, lists, tuples - recurse into contents (don't ref the container)
@@ -705,6 +709,16 @@ def _serialize_result(obj, visited=None):
         return [_serialize_result(v, visited) for v in obj]
     if isinstance(obj, tuple):
         return tuple(_serialize_result(v, visited) for v in obj)
+
+    # Convert numpy scalars to Python primitives for JSON serialization
+    obj_type = type(obj).__name__
+    if obj_type in ('float16', 'float32', 'float64'):
+        return float(obj)
+    if obj_type in ('int8', 'int16', 'int32', 'int64', 'uint8', 'uint16', 'uint32', 'uint64'):
+        return int(obj)
+    if obj_type == 'bool_':
+        return bool(obj)
+
     return obj
 
 def _deserialize_input(obj):