comfy-env 0.0.8__py3-none-any.whl

comfy_env-0.0.8.dist-info/METADATA

@@ -0,0 +1,228 @@
+Metadata-Version: 2.4
+Name: comfy-env
+Version: 0.0.8
+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: 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.8.dist-info/RECORD

@@ -0,0 +1,39 @@
+comfy_env/__init__.py,sha256=U5XtB2UeXZi8JPBhlh0XjQprLsAuVslPoJNWOSvCFhs,4249
+comfy_env/cli.py,sha256=9GvQYrXlJRl0ZaCuFHvRtVxWQ34Axd5Brgu5FWRONp4,11424
+comfy_env/decorator.py,sha256=daFR5aLzshkmo5sRKhSGPcTUgIUWml7Gs6A1bfnDuyc,15718
+comfy_env/errors.py,sha256=egeyXY-j7KpxyA0s67TcJLEfJX23LCAD3v1P4FgQIGE,10917
+comfy_env/install.py,sha256=JaatBM5PyY4Ua6E2HgkLAlFZ2TSEjlI-lAwJ-Hl7wU8,18876
+comfy_env/registry.py,sha256=iZprw2iIAJl-Vvotsf_B_PofzqE-6IiVfNPBHYUYg6g,11577
+comfy_env/resolver.py,sha256=xz7GPlxy02iwwpocIzzbdGnrwnSpi-D5IzpL8SQSgvI,12893
+comfy_env/runner.py,sha256=0YUqzK93u--7pKV6_yVC564AJE9rS3y81t5ZhQi2t4Y,9696
+comfy_env/env/__init__.py,sha256=sybOBrxJCfL4Xry9NNd5xwn9hXIHudXlXDa7SpJkPCE,811
+comfy_env/env/config.py,sha256=fL2P0ScoioPktZEHQnoo1dy-VB5intJEYLHm5fuOmF8,5406
+comfy_env/env/config_file.py,sha256=QLf3WPCqx46v1kVzM4jV_QAz919zmkk0_FRtLD3jO8s,19507
+comfy_env/env/detection.py,sha256=Co8BJmTRCq1ZHDsm6832jF87za0GRAhH7zF04-5QwcE,4949
+comfy_env/env/manager.py,sha256=ysIVlpxRD5x-5X47ESmfly2Vz6jv-9yA6863szHf7-8,24131
+comfy_env/env/security.py,sha256=dNSitAnfBNVdvxgBBntYw33AJaCs_S1MHb7KJhAVYzM,8171
+comfy_env/env/platform/__init__.py,sha256=Nb5MPZIEeanSMEWwqU4p4bnEKTJn1tWcwobnhq9x9IY,614
+comfy_env/env/platform/base.py,sha256=iS0ptTTVjXRwPU4qWUdvHI7jteuzxGSjWr5BUQ7hGiU,2453
+comfy_env/env/platform/darwin.py,sha256=HK3VkLT6DfesAnIXwx2IaUFHTBclF0xTQnC7azWY6Kc,1552
+comfy_env/env/platform/linux.py,sha256=xLp8FEbFqZLQrzIZBI9z3C4g23Ab1ASTHLsXDzsdCoA,2062
+comfy_env/env/platform/windows.py,sha256=nD1-bKU2rGmEJlS-cc5yWXMSA51YQtVupn-lQEO5UYA,14840
+comfy_env/ipc/__init__.py,sha256=pTjgJn5YJxLXmEvuKh3lkCEJQs-6W6_F01jfkFMUi0c,1375
+comfy_env/ipc/bridge.py,sha256=kEy__kco8FVQNj5MyadF5k00YEivcGmifAJAOfr643U,17645
+comfy_env/ipc/protocol.py,sha256=gfWe5yEDUn4QWhcdWFcxn40GqxlW1Uf23j0edOzPPng,7951
+comfy_env/ipc/tensor.py,sha256=DyU28GymKkLPVwzZyKdm2Av222hdaycMgv3KdL5mtO0,12009
+comfy_env/ipc/torch_bridge.py,sha256=WzdwDJa3N_1fEl9OeZxikvMbwryO5P63o0WmEDpS18A,13206
+comfy_env/ipc/transport.py,sha256=XQlRcfQsd4nd909KIYnZKvsS3ksGpGjyVucn8jvmLIU,9698
+comfy_env/ipc/worker.py,sha256=oxTLF9xXrl8CRx_JVNBdkxZh35NuzfkdxhaUtUuXogs,6661
+comfy_env/stubs/__init__.py,sha256=jMeWEKY30y8QqYX9AUyuZbmm607erQTc4N7YaDoAH00,38
+comfy_env/stubs/folder_paths.py,sha256=KEH9ntMH6HOgE595G5dBL9kSTEUvf_shYtQKIyiiDbk,1586
+comfy_env/workers/__init__.py,sha256=IKZwOvrWOGqBLDUIFAalg4CdqzJ_YnAdxo2Ha7gZTJ0,1467
+comfy_env/workers/base.py,sha256=ZILYXlvGCWuCZXmjKqfG8VeD19ihdYaASdlbasl2BMo,2312
+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=DsfxE3LBAWEuGtk-p-YL0UhVQ7VDh73KT_TFRxYN4-Q,12563
+comfy_env/workers/venv.py,sha256=_ekHfZPqBIPY08DjqiXm6cTBQH4DrbxRWR3AAv3mit8,31589
+comfy_env-0.0.8.dist-info/METADATA,sha256=mir9pZymbCPcnI6yc1YrjxyMMnX3qm3pUR7NILB5iCE,5371
+comfy_env-0.0.8.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+comfy_env-0.0.8.dist-info/entry_points.txt,sha256=J4fXeqgxU_YenuW_Zxn_pEL7J-3R0--b6MS5t0QmAr0,49
+comfy_env-0.0.8.dist-info/licenses/LICENSE,sha256=E68QZMMpW4P2YKstTZ3QU54HRQO8ecew09XZ4_Vn870,1093
+comfy_env-0.0.8.dist-info/RECORD,,

comfy_env-0.0.8.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

comfy_env-0.0.8.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+comfy-env = comfy_env.cli:main

comfy_env-0.0.8.dist-info/licenses/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.