@@ -181,24 +181,25 @@ def run_command_logic(args):
181
181
182
182
benchmark_output_dict ["results-comparison" ] = {}
183
183
for step in run_stages :
184
- for run_name , result_run in benchmark_output_dict [step ].items ():
185
- step_df_dict [step ]["df_dict" ]["run-name" ].append (run_name )
186
- for pos , metric_json_path in enumerate (step_df_dict [step ]["metric_json_path" ]):
187
- metric_name = step_df_dict [step ]["sorting_metric_names" ][pos ]
188
- metric_value = None
189
- try :
190
- metric_value = findJsonPath (metric_json_path , result_run )
191
- except KeyError :
192
- print (
193
- "Error retrieving {} metric from JSON PATH {} on file {}" .format (metric_name , metric_json_path ,
194
- run_name ))
195
- pass
196
- step_df_dict [step ]["df_dict" ][metric_name ].append (metric_value )
197
- resultsDataFrame = pd .DataFrame (step_df_dict [step ]["df_dict" ])
198
- resultsDataFrame .sort_values (step_df_dict [step ]["sorting_metric_names" ],
199
- ascending = step_df_dict [step ]["sorting_metric_sorting_direction" ], inplace = True )
200
- benchmark_output_dict ["key-results" ][step ] = from_resultsDF_to_key_results_dict (resultsDataFrame , step ,
201
- step_df_dict )
184
+ if run_only_steps is not None and step not in run_only_steps :
185
+ for run_name , result_run in benchmark_output_dict [step ].items ():
186
+ step_df_dict [step ]["df_dict" ]["run-name" ].append (run_name )
187
+ for pos , metric_json_path in enumerate (step_df_dict [step ]["metric_json_path" ]):
188
+ metric_name = step_df_dict [step ]["sorting_metric_names" ][pos ]
189
+ metric_value = None
190
+ try :
191
+ metric_value = findJsonPath (metric_json_path , result_run )
192
+ except KeyError :
193
+ print (
194
+ "Error retrieving {} metric from JSON PATH {} on file {}" .format (metric_name , metric_json_path ,
195
+ run_name ))
196
+ pass
197
+ step_df_dict [step ]["df_dict" ][metric_name ].append (metric_value )
198
+ resultsDataFrame = pd .DataFrame (step_df_dict [step ]["df_dict" ])
199
+ resultsDataFrame .sort_values (step_df_dict [step ]["sorting_metric_names" ],
200
+ ascending = step_df_dict [step ]["sorting_metric_sorting_direction" ], inplace = True )
201
+ benchmark_output_dict ["key-results" ][step ] = from_resultsDF_to_key_results_dict (resultsDataFrame , step ,
202
+ step_df_dict )
202
203
#####################
203
204
# Run Info Metadata #
204
205
#####################
0 commit comments