diff --git a/cluster4npu_ui/core/functions/Multidongle.py b/cluster4npu_ui/core/functions/Multidongle.py index 5ebed2a..3031438 100644 --- a/cluster4npu_ui/core/functions/Multidongle.py +++ b/cluster4npu_ui/core/functions/Multidongle.py @@ -426,7 +426,7 @@ class MultiDongle: retrieval_successful = False break - if retrieval_successful and inf_node_output_list: + if retrieval_successful and len(inf_node_output_list) > 0: # Process output nodes if output_descriptor.header.num_output_node == 1: raw_output_array = inf_node_output_list[0].flatten() @@ -690,7 +690,7 @@ def postprocess(raw_model_output: list) -> float: Post-processes the raw model output. Assumes the model output is a list/array where the first element is the desired probability. """ - if raw_model_output and len(raw_model_output) > 0: + if raw_model_output is not None and len(raw_model_output) > 0: probability = raw_model_output[0] return float(probability) return 0.0 # Default or error value