@@ -119,9 +119,9 @@ def get_ready(self):
119
119
if self .has_result (key ) or cs :
120
120
continue
121
121
node = self ._working_graph [key ]
122
- if self .taskref (node ):
122
+ if DaskVineDag .taskref (node ):
123
123
rs .update (self .set_result (key , self .get_result (node .key )))
124
- elif self .symbolp (node ):
124
+ elif DaskVineDag .symbolp (node ):
125
125
rs .update (self .set_result (key , node ))
126
126
else :
127
127
rs [key ] = node
@@ -147,11 +147,11 @@ def set_result(self, key, value):
147
147
continue
148
148
149
149
node = self ._working_graph [p ]
150
- if self .taskref (node ):
150
+ if DaskVineDag .taskref (node ):
151
151
rs .update (
152
- self .set_result (p , self .get_result (node ))
152
+ self .set_result (p , self .get_result (node . key ))
153
153
) # case e.g, "x": "y", and we just set the value of "y"
154
- elif self .symbolp (node ):
154
+ elif DaskVineDag .symbolp (node ):
155
155
rs .update (self .set_result (p , node ))
156
156
else :
157
157
rs [p ] = node
@@ -169,14 +169,14 @@ def set_result(self, key, value):
169
169
170
170
def _add_second_targets (self , key ):
171
171
v = self ._working_graph [key ]
172
- if self .taskref (v ):
172
+ if DaskVineDag .taskref (v ):
173
173
lst = [v ]
174
174
elif DaskVineDag .containerp (v ):
175
175
lst = v
176
176
else :
177
177
return
178
178
for c in lst :
179
- if self .taskref (c ):
179
+ if DaskVineDag .taskref (c ):
180
180
self ._targets .add (c .key )
181
181
self ._add_second_targets (c .key )
182
182
0 commit comments