diff --git a/osbenchmark/aggregator.py b/osbenchmark/aggregator.py index 58b0682a..016d90f4 100644 --- a/osbenchmark/aggregator.py +++ b/osbenchmark/aggregator.py @@ -23,7 +23,7 @@ def __init__(self, cfg, test_executions_dict, args): def count_iterations_for_each_op(self, test_execution) -> None: matching_test_procedure = next((tp for tp in self.loaded_workload.test_procedures if tp.name == self.test_procedure_name), None) - workload_params = getattr(test_execution, 'workload_params', {}) + workload_params = test_execution.workload_params if test_execution.workload_params else {} test_execution_id = test_execution.test_execution_id self.accumulated_iterations[test_execution_id] = {}