Skip to content

Commit 5960f94

Browse files
Move a few files from comfy -> comfy_execution.
Python code in the comfy folder should not import things from outside it.
1 parent 5cfe38f commit 5960f94

File tree

8 files changed

+11
-12
lines changed

8 files changed

+11
-12
lines changed

comfy/caching.py renamed to comfy_execution/caching.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
import itertools
22
from typing import Sequence, Mapping
3-
from comfy.graph import DynamicPrompt
3+
from comfy_execution.graph import DynamicPrompt
44

55
import nodes
66

7-
from comfy.graph_utils import is_link
7+
from comfy_execution.graph_utils import is_link
88

99
class CacheKeySet:
1010
def __init__(self, dynprompt, node_ids, is_changed_cache):

comfy/graph.py renamed to comfy_execution/graph.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import nodes
22

3-
from comfy.graph_utils import is_link
3+
from comfy_execution.graph_utils import is_link
44

55
class DependencyCycleError(Exception):
66
pass
File renamed without changes.

execution.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,9 @@
1313
import nodes
1414

1515
import comfy.model_management
16-
import comfy.graph_utils
17-
from comfy.graph import get_input_info, ExecutionList, DynamicPrompt, ExecutionBlocker
18-
from comfy.graph_utils import is_link, GraphBuilder
19-
from comfy.caching import HierarchicalCache, LRUCache, CacheKeySetInputSignature, CacheKeySetID
16+
from comfy_execution.graph import get_input_info, ExecutionList, DynamicPrompt, ExecutionBlocker
17+
from comfy_execution.graph_utils import is_link, GraphBuilder
18+
from comfy_execution.caching import HierarchicalCache, LRUCache, CacheKeySetInputSignature, CacheKeySetID
2019
from comfy.cli_args import args
2120

2221
class ExecutionResult(Enum):

tests/inference/test_execution.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
import urllib.request
1414
import urllib.parse
1515
import urllib.error
16-
from comfy.graph_utils import GraphBuilder, Node
16+
from comfy_execution.graph_utils import GraphBuilder, Node
1717

1818
class RunResult:
1919
def __init__(self, prompt_id: str):

tests/inference/testing_nodes/testing-pack/flow_control.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
from comfy.graph_utils import GraphBuilder, is_link
2-
from comfy.graph import ExecutionBlocker
1+
from comfy_execution.graph_utils import GraphBuilder, is_link
2+
from comfy_execution.graph import ExecutionBlocker
33
from .tools import VariantSupport
44

55
NUM_FLOW_SOCKETS = 5

tests/inference/testing_nodes/testing-pack/specific_tests.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import torch
22
from .tools import VariantSupport
3-
from comfy.graph_utils import GraphBuilder
3+
from comfy_execution.graph_utils import GraphBuilder
44

55
class TestLazyMixImages:
66
@classmethod

tests/inference/testing_nodes/testing-pack/util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from comfy.graph_utils import GraphBuilder
1+
from comfy_execution.graph_utils import GraphBuilder
22
from .tools import VariantSupport
33

44
@VariantSupport()

0 commit comments

Comments
 (0)