From 7cde7443b0059e17535ce4dfcfd781fee5f36e24 Mon Sep 17 00:00:00 2001 From: "Dr.Lt.Data" Date: Sat, 17 Jun 2023 16:10:49 +0900 Subject: [PATCH] robust patch for None return --- execution.py | 2 +- worklist_execution.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/execution.py b/execution.py index a8e04333f..bc7b57907 100644 --- a/execution.py +++ b/execution.py @@ -90,7 +90,7 @@ def get_output_data(obj, input_data_all): results.append(r) output = [] - if len(results) > 0: + if len(results) > 0 and results[0] is not None: # check which outputs need concatenating output_is_list = [False] * len(results[0]) if hasattr(obj, "OUTPUT_IS_LIST"): diff --git a/worklist_execution.py b/worklist_execution.py index 6fa62cce3..e28170a62 100644 --- a/worklist_execution.py +++ b/worklist_execution.py @@ -110,7 +110,9 @@ def is_incomplete_input_slots(class_def, inputs, outputs): if isinstance(input_data, list): input_unique_id = input_data[0] - if input_unique_id not in outputs or outputs[input_unique_id][input_data[1]] == [None]: + if input_unique_id not in outputs or \ + len(outputs[input_unique_id]) == 0 or \ + outputs[input_unique_id][input_data[1]] == [None]: return True return False