comfy-env 0.0.36__tar.gz

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

@@ -0,0 +1,67 @@
+name: Bump Version & Publish
+
+on:
+  push:
+    branches:
+      - main
+  workflow_dispatch:
+
+jobs:
+  bump-and-publish:
+    runs-on: ubuntu-latest
+    environment: pypi
+    permissions:
+      contents: write
+      id-token: write
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          token: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Setup Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.11'
+
+      - name: Get current version
+        id: get_version
+        run: |
+          current=$(grep '^version = ' pyproject.toml | sed 's/version = "\(.*\)"/\1/' | tr -d '\r')
+          echo "current=$current" >> $GITHUB_OUTPUT
+
+      - name: Bump patch version
+        id: bump_version
+        run: |
+          current="${{ steps.get_version.outputs.current }}"
+          IFS='.' read -r major minor patch <<< "$current"
+          new_patch=$((patch + 1))
+          new_version="${major}.${minor}.${new_patch}"
+          sed -i "s/^version = \".*\"/version = \"${new_version}\"/" pyproject.toml
+          echo "new_version=$new_version" >> $GITHUB_OUTPUT
+          echo "Bumped version: $current -> $new_version"
+
+      - name: Commit version bump
+        run: |
+          git config user.name "github-actions[bot]"
+          git config user.email "github-actions[bot]@users.noreply.github.com"
+          git add pyproject.toml
+          git commit -m "Bump version to ${{ steps.bump_version.outputs.new_version }} [skip ci]"
+          git tag "v${{ steps.bump_version.outputs.new_version }}"
+          git push origin main --tags
+
+      - name: Install build tools
+        run: pip install build
+
+      - name: Build package
+        run: python -m build
+
+      - name: Create GitHub Release
+        uses: softprops/action-gh-release@v1
+        with:
+          tag_name: v${{ steps.bump_version.outputs.new_version }}
+          name: v${{ steps.bump_version.outputs.new_version }}
+          generate_release_notes: true
+          files: dist/*
+
+      - name: Publish to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1

comfy_env-0.0.36/.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.36/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.36/PKG-INFO

@@ -0,0 +1,229 @@
+Metadata-Version: 2.4
+Name: comfy-env
+Version: 0.0.36
+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
+```
+
+## Configurations
+
+### 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.36/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
+```
+
+## Configurations
+
+### 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.36/pyproject.toml

@@ -0,0 +1,50 @@
+[project]
+name = "comfy-env"
+version = "0.0.36"
+description = "Environment management for ComfyUI custom nodes - CUDA wheel resolution and process isolation"
+readme = "README.md"
+license = {text = "MIT"}
+requires-python = ">=3.10"
+authors = [
+    {name = "Andrea Pozzetti"}
+]
+keywords = ["comfyui", "cuda", "wheels", "environment", "isolation", "venv", "process"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+]
+dependencies = [
+    "tomli>=2.0.0; python_version < '3.11'",  # TOML parsing (built-in tomllib for 3.11+)
+    "uv>=0.4.0",  # Fast Python package installer and venv creator
+    "pyyaml>=6.0",  # YAML parsing for wheel_sources.yml
+]
+
+[project.optional-dependencies]
+dev = ["pytest", "ruff", "mypy"]
+
+[project.urls]
+Homepage = "https://github.com/PozzettiAndrea/comfy-env"
+Repository = "https://github.com/PozzettiAndrea/comfy-env"
+Issues = "https://github.com/PozzettiAndrea/comfy-env/issues"
+
+[project.scripts]
+comfy-env = "comfy_env.cli:main"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/comfy_env"]
+
+[tool.hatch.build.targets.wheel.force-include]
+"src/comfy_env/wheel_sources.yml" = "comfy_env/wheel_sources.yml"
+
+[tool.ruff]
+line-length = 100
+target-version = "py310"