Spaces:
Running
Running
Fix "n.id in _ops" check.
Browse files
lynxkite-core/src/lynxkite/core/workspace.py
CHANGED
@@ -102,12 +102,14 @@ class Workspace(BaseConfig):
|
|
102 |
return self
|
103 |
catalog = ops.CATALOGS[self.env]
|
104 |
_ops = {n.id: catalog[n.data.title] for n in self.nodes if n.data.title in catalog}
|
105 |
-
valid_targets = set(
|
106 |
-
|
107 |
-
|
108 |
-
|
109 |
-
|
110 |
-
|
|
|
|
|
111 |
edges = [
|
112 |
edge
|
113 |
for edge in self.edges
|
|
|
102 |
return self
|
103 |
catalog = ops.CATALOGS[self.env]
|
104 |
_ops = {n.id: catalog[n.data.title] for n in self.nodes if n.data.title in catalog}
|
105 |
+
valid_targets = set()
|
106 |
+
valid_sources = set()
|
107 |
+
for n in self.nodes:
|
108 |
+
if n.id in _ops:
|
109 |
+
for h in _ops[n.id].inputs:
|
110 |
+
valid_targets.add((n.id, h))
|
111 |
+
for h in _ops[n.id].outputs:
|
112 |
+
valid_sources.add((n.id, h))
|
113 |
edges = [
|
114 |
edge
|
115 |
for edge in self.edges
|