Fixed style

intel · Jun 18, 2024 · bdb3f81 · bdb3f81
1 parent 9b5f008
commit bdb3f81
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/scripts/build_report.py b/scripts/build_report.py
@@ -33,7 +33,7 @@ def transform_df(df, param_cols, tflops_col, hbm_col, benchmark, compiler):
     # Build json with parameters
     df_results = pd.DataFrame()
     df_results["params"] = [json.dumps(j) for j in df[param_cols].astype(int).to_dict("records")]
-    df_results['tflops']  = df[tflops_col]
+    df_results['tflops'] = df[tflops_col]
     if hbm_col is not None:
         df_results['hbm_gbs'] = df[hbm_col]
 
@@ -57,10 +57,8 @@ def main():
     args = parse_args()
     param_cols = args.param_cols.split(",")
     df = pd.read_csv(args.source)
-    result_df = transform_df(
-        df, param_cols=param_cols, tflops_col=args.tflops_col,
-        hbm_col=args.hbm_col, benchmark=args.benchmark, compiler=args.compiler
-    )
+    result_df = transform_df(df, param_cols=param_cols, tflops_col=args.tflops_col, hbm_col=args.hbm_col,
+                             benchmark=args.benchmark, compiler=args.compiler)
     result_df.to_csv(args.target, index=False)