comfy-env 0.0.18__tar.gz

comfy_env-0.0.18/.github/workflows/publish.yml

@@ -0,0 +1,28 @@
+name: Publish to PyPI
+
+on:
+  release:
+    types: [published]
+
+jobs:
+  publish:
+    runs-on: ubuntu-latest
+    environment: pypi
+    permissions:
+      id-token: write  # Required for trusted publishing
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+
+      - name: Install build tools
+        run: pip install build
+
+      - name: Build package
+        run: python -m build
+
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1

comfy_env-0.0.18/.gitignore

@@ -0,0 +1,32 @@
+# Python
+__pycache__/
+*.py[cod]
+*.so
+*.egg
+*.egg-info/
+dist/
+build/
+.eggs/
+
+# Virtual environments
+.venv/
+venv/
+ENV/
+
+# IDE
+.idea/
+.vscode/
+*.swp
+*.swo
+
+# Jupyter
+.ipynb_checkpoints/
+
+# Testing
+.pytest_cache/
+.coverage
+htmlcov/
+
+# OS
+.DS_Store
+Thumbs.db

comfy_env-0.0.18/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Andrea Pozzetti
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

comfy_env-0.0.18/PKG-INFO

@@ -0,0 +1,229 @@
+Metadata-Version: 2.4
+Name: comfy-env
+Version: 0.0.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
+Project-URL: Issues, https://github.com/PozzettiAndrea/comfy-env/issues
+Author: Andrea Pozzetti
+License: MIT
+License-File: LICENSE
+Keywords: comfyui,cuda,environment,isolation,process,venv,wheels
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: >=3.10
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: tomli>=2.0.0; python_version < '3.11'
+Requires-Dist: uv>=0.4.0
+Provides-Extra: dev
+Requires-Dist: mypy; extra == 'dev'
+Requires-Dist: pytest; extra == 'dev'
+Requires-Dist: ruff; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# comfy-env
+
+Environment management for ComfyUI custom nodes. Provides:
+
+1. **CUDA Wheel Resolution** - Install pre-built CUDA wheels (nvdiffrast, pytorch3d) without compilation
+2. **Process Isolation** - Run nodes in separate Python environments with their own dependencies
+
+## Why?
+
+ComfyUI custom nodes face two challenges:
+
+**Type 1: Dependency Conflicts**
+- Node A needs `torch==2.1.0` with CUDA 11.8
+- Node B needs `torch==2.8.0` with CUDA 12.8
+
+**Type 2: CUDA Package Installation**
+- Users don't have compilers installed
+- Building from source takes forever
+- pip install fails with cryptic errors
+
+This package solves both problems.
+
+## Installation
+
+```bash
+pip install comfy-env
+```
+
+Requires [uv](https://github.com/astral-sh/uv) for fast environment creation:
+
+```bash
+curl -LsSf https://astral.sh/uv/install.sh | sh
+```
+
+## Quick Start
+
+### In-Place Installation (Type 2 - CUDA Wheels)
+
+Create a `comfy-env.toml` in your node directory:
+
+```toml
+[env]
+name = "my-node"
+python = "3.10"
+cuda = "auto"
+
+[packages]
+requirements = ["transformers>=4.56", "pillow"]
+no_deps = ["nvdiffrast==0.4.0", "pytorch3d>=0.7.8"]
+
+[sources]
+wheel_sources = ["https://github.com/PozzettiAndrea/nvdiffrast-full-wheels/releases/download/"]
+```
+
+Then in your `__init__.py`:
+
+```python
+from comfy_env import install
+
+# Install CUDA wheels into current environment
+install()
+```
+
+### Process Isolation (Type 1 - Separate Venv)
+
+For nodes that need completely separate dependencies:
+
+```python
+from comfy_env import isolated
+
+@isolated(env="my-node")
+class MyNode:
+    FUNCTION = "process"
+    RETURN_TYPES = ("IMAGE",)
+
+    def process(self, image):
+        # Runs in isolated subprocess with its own venv
+        import conflicting_package
+        return (result,)
+```
+
+## CLI
+
+```bash
+# Show detected environment
+comfy-env info
+
+# Install from config
+comfy-env install
+
+# Dry run (show what would be installed)
+comfy-env install --dry-run
+
+# Resolve wheel URLs without installing
+comfy-env resolve nvdiffrast==0.4.0
+
+# Verify installation
+comfy-env doctor
+```
+
+## Configuration
+
+### comfy-env.toml
+
+```toml
+[env]
+name = "my-node"          # Unique name for caching
+python = "3.10"           # Python version
+cuda = "auto"             # "auto", "12.8", "12.4", or null
+
+[packages]
+requirements = [          # Regular pip packages
+    "transformers>=4.56",
+    "pillow",
+]
+no_deps = [               # CUDA packages (installed with --no-deps)
+    "nvdiffrast==0.4.0",
+    "pytorch3d>=0.7.8",
+]
+
+[sources]
+wheel_sources = [         # GitHub releases with pre-built wheels
+    "https://github.com/.../releases/download/",
+]
+index_urls = [            # Extra pip index URLs
+    "https://pypi.org/simple/",
+]
+
+[worker]                  # For isolation mode
+package = "worker"        # worker/__main__.py
+```
+
+### Template Variables
+
+Wheel URLs support these template variables:
+
+| Variable | Example | Description |
+|----------|---------|-------------|
+| `{cuda_version}` | `12.8` | Full CUDA version |
+| `{cuda_short}` | `128` | CUDA without dot |
+| `{torch_version}` | `2.8.0` | PyTorch version |
+| `{torch_mm}` | `28` | PyTorch major.minor |
+| `{py_version}` | `3.10` | Python version |
+| `{py_short}` | `310` | Python without dot |
+| `{platform}` | `linux_x86_64` | Platform tag |
+
+## API Reference
+
+### install()
+
+```python
+from comfy_env import install
+
+# Auto-discover config
+install()
+
+# Explicit config
+install(config="comfy-env.toml")
+
+# Isolated mode (creates separate venv)
+install(mode="isolated")
+
+# Dry run
+install(dry_run=True)
+```
+
+### WheelResolver
+
+```python
+from comfy_env import RuntimeEnv, WheelResolver
+
+env = RuntimeEnv.detect()
+resolver = WheelResolver()
+
+url = resolver.resolve("nvdiffrast", "0.4.0", env)
+print(url)  # https://github.com/.../nvdiffrast-0.4.0+cu128torch28-...whl
+```
+
+### Workers (for isolation)
+
+```python
+from comfy_env import TorchMPWorker
+
+# Same-venv isolation (zero-copy tensors)
+worker = TorchMPWorker()
+result = worker.call(my_function, image=tensor)
+```
+
+## GPU Detection
+
+```python
+from comfy_env import detect_cuda_version, get_gpu_summary
+
+cuda = detect_cuda_version()  # "12.8", "12.4", or None
+print(get_gpu_summary())
+# GPU 0: NVIDIA GeForce RTX 5090 (sm_120) [Blackwell - CUDA 12.8]
+```
+
+## License
+
+MIT - see LICENSE file.

comfy_env-0.0.18/README.md

@@ -0,0 +1,201 @@
+# comfy-env
+
+Environment management for ComfyUI custom nodes. Provides:
+
+1. **CUDA Wheel Resolution** - Install pre-built CUDA wheels (nvdiffrast, pytorch3d) without compilation
+2. **Process Isolation** - Run nodes in separate Python environments with their own dependencies
+
+## Why?
+
+ComfyUI custom nodes face two challenges:
+
+**Type 1: Dependency Conflicts**
+- Node A needs `torch==2.1.0` with CUDA 11.8
+- Node B needs `torch==2.8.0` with CUDA 12.8
+
+**Type 2: CUDA Package Installation**
+- Users don't have compilers installed
+- Building from source takes forever
+- pip install fails with cryptic errors
+
+This package solves both problems.
+
+## Installation
+
+```bash
+pip install comfy-env
+```
+
+Requires [uv](https://github.com/astral-sh/uv) for fast environment creation:
+
+```bash
+curl -LsSf https://astral.sh/uv/install.sh | sh
+```
+
+## Quick Start
+
+### In-Place Installation (Type 2 - CUDA Wheels)
+
+Create a `comfy-env.toml` in your node directory:
+
+```toml
+[env]
+name = "my-node"
+python = "3.10"
+cuda = "auto"
+
+[packages]
+requirements = ["transformers>=4.56", "pillow"]
+no_deps = ["nvdiffrast==0.4.0", "pytorch3d>=0.7.8"]
+
+[sources]
+wheel_sources = ["https://github.com/PozzettiAndrea/nvdiffrast-full-wheels/releases/download/"]
+```
+
+Then in your `__init__.py`:
+
+```python
+from comfy_env import install
+
+# Install CUDA wheels into current environment
+install()
+```
+
+### Process Isolation (Type 1 - Separate Venv)
+
+For nodes that need completely separate dependencies:
+
+```python
+from comfy_env import isolated
+
+@isolated(env="my-node")
+class MyNode:
+    FUNCTION = "process"
+    RETURN_TYPES = ("IMAGE",)
+
+    def process(self, image):
+        # Runs in isolated subprocess with its own venv
+        import conflicting_package
+        return (result,)
+```
+
+## CLI
+
+```bash
+# Show detected environment
+comfy-env info
+
+# Install from config
+comfy-env install
+
+# Dry run (show what would be installed)
+comfy-env install --dry-run
+
+# Resolve wheel URLs without installing
+comfy-env resolve nvdiffrast==0.4.0
+
+# Verify installation
+comfy-env doctor
+```
+
+## Configuration
+
+### comfy-env.toml
+
+```toml
+[env]
+name = "my-node"          # Unique name for caching
+python = "3.10"           # Python version
+cuda = "auto"             # "auto", "12.8", "12.4", or null
+
+[packages]
+requirements = [          # Regular pip packages
+    "transformers>=4.56",
+    "pillow",
+]
+no_deps = [               # CUDA packages (installed with --no-deps)
+    "nvdiffrast==0.4.0",
+    "pytorch3d>=0.7.8",
+]
+
+[sources]
+wheel_sources = [         # GitHub releases with pre-built wheels
+    "https://github.com/.../releases/download/",
+]
+index_urls = [            # Extra pip index URLs
+    "https://pypi.org/simple/",
+]
+
+[worker]                  # For isolation mode
+package = "worker"        # worker/__main__.py
+```
+
+### Template Variables
+
+Wheel URLs support these template variables:
+
+| Variable | Example | Description |
+|----------|---------|-------------|
+| `{cuda_version}` | `12.8` | Full CUDA version |
+| `{cuda_short}` | `128` | CUDA without dot |
+| `{torch_version}` | `2.8.0` | PyTorch version |
+| `{torch_mm}` | `28` | PyTorch major.minor |
+| `{py_version}` | `3.10` | Python version |
+| `{py_short}` | `310` | Python without dot |
+| `{platform}` | `linux_x86_64` | Platform tag |
+
+## API Reference
+
+### install()
+
+```python
+from comfy_env import install
+
+# Auto-discover config
+install()
+
+# Explicit config
+install(config="comfy-env.toml")
+
+# Isolated mode (creates separate venv)
+install(mode="isolated")
+
+# Dry run
+install(dry_run=True)
+```
+
+### WheelResolver
+
+```python
+from comfy_env import RuntimeEnv, WheelResolver
+
+env = RuntimeEnv.detect()
+resolver = WheelResolver()
+
+url = resolver.resolve("nvdiffrast", "0.4.0", env)
+print(url)  # https://github.com/.../nvdiffrast-0.4.0+cu128torch28-...whl
+```
+
+### Workers (for isolation)
+
+```python
+from comfy_env import TorchMPWorker
+
+# Same-venv isolation (zero-copy tensors)
+worker = TorchMPWorker()
+result = worker.call(my_function, image=tensor)
+```
+
+## GPU Detection
+
+```python
+from comfy_env import detect_cuda_version, get_gpu_summary
+
+cuda = detect_cuda_version()  # "12.8", "12.4", or None
+print(get_gpu_summary())
+# GPU 0: NVIDIA GeForce RTX 5090 (sm_120) [Blackwell - CUDA 12.8]
+```
+
+## License
+
+MIT - see LICENSE file.

comfy_env-0.0.18/examples/basic_node/__init__.py

@@ -0,0 +1,5 @@
+"""Example ComfyUI node using process isolation."""
+
+from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+
+__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]

comfy_env-0.0.18/examples/basic_node/comfy-env.toml

@@ -0,0 +1,65 @@
+# Example isolation configuration for a ComfyUI custom node.
+#
+# This file defines the isolated Python environment for your node.
+# The comfyui-isolation library will auto-discover this file.
+#
+# Usage in your node:
+#     bridge = WorkerBridge.from_config_file(
+#         node_dir=Path(__file__).parent,
+#         worker_script=Path(__file__).parent / "worker.py",
+#     )
+
+[env]
+# Unique name for this environment (used for caching)
+name = "example-node"
+
+# Python version
+python = "3.10"
+
+# CUDA version: "auto" (detect), "12.4", "12.8", or null (CPU only)
+cuda = "auto"
+
+# Optional: specific PyTorch version (if not specified, latest from index is used)
+# pytorch_version = "2.4.1"
+
+
+[packages]
+# Required packages
+requirements = [
+    "torch",
+    "torchvision",
+    "pillow",
+    "numpy",
+]
+
+# Optional: path to additional requirements.txt (relative to this file's directory)
+# requirements_file = "requirements.txt"
+
+
+[sources]
+# Custom wheel repositories (passed as --find-links to pip)
+# wheel_sources = [
+#     "https://my-custom-wheels.github.io/",
+# ]
+
+# Extra package indexes (passed as --extra-index-url to pip)
+# index_urls = [
+#     "https://test.pypi.org/simple/",
+# ]
+
+
+[variables]
+# Define variables for substitution in requirements.
+# You can reference these with {variable_name} syntax.
+#
+# Auto-populated variables (no need to define):
+#   {cuda_version} - e.g., "12.4"
+#   {cuda_short} - e.g., "124"
+#   {pytorch_version} - e.g., "2.9.1"
+#   {pytorch_short} - e.g., "291" (version without dots)
+#   {pytorch_mm} - e.g., "29" (major.minor without dot, for wheel names)
+#
+# Example:
+#   pytorch_version = "2.4.1"
+#   pytorch3d_version = "0.7.8+5043d15pt{pytorch_version}cu{cuda_short}"
+#   flex_gemm = "0.0.1+cu{cuda_short}torch{pytorch_mm}"