Skip to content

Commit ded73b1

Browse files
[FIX] Handle None response model in fetch_outputs_with_tree
When the trace server returns a non-200 status for both datatree and timegraph tree endpoints, response.model is None. Accessing .model on None caused an AttributeError, failing CI tests.
1 parent e76ab3f commit ded73b1

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

tmll/tmll_client.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -265,6 +265,10 @@ def fetch_outputs_with_tree(self, experiment: Experiment, custom_output_ids: Opt
265265
self.logger.warning(f"Output type '{output.type}' is not supported.")
266266
continue
267267

268+
if response.model is None:
269+
self.logger.warning(f"Tree of the output '{output.name}' is None.")
270+
continue
271+
268272
model = response.model.model
269273
if model is None:
270274
self.logger.warning(f"Tree of the output '{output.name}' is None.")

0 commit comments

Comments
 (0)