darabos commited on
Commit
bcfe577
·
1 Parent(s): 937bb99

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
- (n.id, h) for n in self.nodes for h in _ops[n.id].inputs if n.id in _ops
107
- )
108
- valid_sources = set(
109
- (n.id, h) for n in self.nodes for h in _ops[n.id].outputs if n.id in _ops
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