Move a few files from comfy -> comfy_execution.

Python code in the comfy folder should not import things from outside it.
This commit is contained in:
comfyanonymous 2024-08-15 09:37:30 -04:00
parent 5cfe38f41c
commit 5960f946a9
8 changed files with 11 additions and 12 deletions

View File

@ -1,10 +1,10 @@
import itertools import itertools
from typing import Sequence, Mapping from typing import Sequence, Mapping
from comfy.graph import DynamicPrompt from comfy_execution.graph import DynamicPrompt
import nodes import nodes
from comfy.graph_utils import is_link from comfy_execution.graph_utils import is_link
class CacheKeySet: class CacheKeySet:
def __init__(self, dynprompt, node_ids, is_changed_cache): def __init__(self, dynprompt, node_ids, is_changed_cache):

View File

@ -1,6 +1,6 @@
import nodes import nodes
from comfy.graph_utils import is_link from comfy_execution.graph_utils import is_link
class DependencyCycleError(Exception): class DependencyCycleError(Exception):
pass pass

View File

@ -13,10 +13,9 @@ import torch
import nodes import nodes
import comfy.model_management import comfy.model_management
import comfy.graph_utils from comfy_execution.graph import get_input_info, ExecutionList, DynamicPrompt, ExecutionBlocker
from comfy.graph import get_input_info, ExecutionList, DynamicPrompt, ExecutionBlocker from comfy_execution.graph_utils import is_link, GraphBuilder
from comfy.graph_utils import is_link, GraphBuilder from comfy_execution.caching import HierarchicalCache, LRUCache, CacheKeySetInputSignature, CacheKeySetID
from comfy.caching import HierarchicalCache, LRUCache, CacheKeySetInputSignature, CacheKeySetID
from comfy.cli_args import args from comfy.cli_args import args
class ExecutionResult(Enum): class ExecutionResult(Enum):

View File

@ -13,7 +13,7 @@ import uuid
import urllib.request import urllib.request
import urllib.parse import urllib.parse
import urllib.error import urllib.error
from comfy.graph_utils import GraphBuilder, Node from comfy_execution.graph_utils import GraphBuilder, Node
class RunResult: class RunResult:
def __init__(self, prompt_id: str): def __init__(self, prompt_id: str):

View File

@ -1,5 +1,5 @@
from comfy.graph_utils import GraphBuilder, is_link from comfy_execution.graph_utils import GraphBuilder, is_link
from comfy.graph import ExecutionBlocker from comfy_execution.graph import ExecutionBlocker
from .tools import VariantSupport from .tools import VariantSupport
NUM_FLOW_SOCKETS = 5 NUM_FLOW_SOCKETS = 5

View File

@ -1,6 +1,6 @@
import torch import torch
from .tools import VariantSupport from .tools import VariantSupport
from comfy.graph_utils import GraphBuilder from comfy_execution.graph_utils import GraphBuilder
class TestLazyMixImages: class TestLazyMixImages:
@classmethod @classmethod

View File

@ -1,4 +1,4 @@
from comfy.graph_utils import GraphBuilder from comfy_execution.graph_utils import GraphBuilder
from .tools import VariantSupport from .tools import VariantSupport
@VariantSupport() @VariantSupport()