format it

LSC-Unicamp · Dec 3, 2024 · 4438e18 · 4438e18
1 parent c951310
commit 4438e18
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/spinner/runner/utilities.py b/spinner/runner/utilities.py
@@ -23,7 +23,9 @@ def run_benchmarks(config, output, extra_args):
     # Capture specified environment variables
     if "os_envs" in bench_metadata:
         env_vars_to_capture = bench_metadata["os_envs"]
-        bench_metadata["start_env"] = {var: os.environ.get(var) for var in env_vars_to_capture}
+        bench_metadata["start_env"] = {
+            var: os.environ.get(var) for var in env_vars_to_capture
+        }
     else:
         bench_metadata["start_env"] = {}
 
@@ -102,11 +104,12 @@ def _update_progress(
     # Capture specified environment variables at the end
     if "os_envs" in bench_metadata:
         env_vars_to_capture = bench_metadata["os_envs"]
-        bench_metadata["end_env"] = {var: os.environ.get(var) for var in env_vars_to_capture}
+        bench_metadata["end_env"] = {
+            var: os.environ.get(var) for var in env_vars_to_capture
+        }
     else:
         bench_metadata["end_env"] = {}
 
     rprint(execution_df)
     with open(output, "wb") as f:
         pickle.dump({"metadata": bench_metadata, "dataframe": execution_df}, f)
-