Skip to content

Commit 56283ca

Browse files
fix: solve the missing key in results (#31)
1 parent 898c6da commit 56283ca

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

modzy/results.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -177,15 +177,15 @@ def get_source_outputs(self, source_name):
177177
if source_name in source: # deal with legacy double nesting of source source_name
178178
source = source[source_name]
179179
return source
180-
except KeyError:
180+
except (KeyError, AttributeError):
181181
pass
182182

183183
try:
184184
source = self.failures[source_name]
185185
if source_name in source: # deal with legacy double nesting of source source_name
186186
source = source[source_name]
187187
raise ResultsError(source.error)
188-
except KeyError:
188+
except (KeyError, AttributeError):
189189
pass
190190

191191
# TODO: can we give a better error message if job canceled?
@@ -218,12 +218,12 @@ def get_first_outputs(self):
218218
def _get_first_source_name(self):
219219
try:
220220
return next(iter(self.results))
221-
except StopIteration:
221+
except (StopIteration, AttributeError):
222222
pass
223223

224224
try:
225225
return next(iter(self.failures))
226-
except StopIteration:
226+
except (StopIteration, AttributeError):
227227
pass
228228

229229
# TODO: can we give a better error message if job canceled?

0 commit comments

Comments
 (0)