pre-commit auto-fixes

autoatml · Jan 6, 2025 · 5b05b43 · 5b05b43
1 parent dd0911a
commit 5b05b43
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/autoplex/auto/phonons/jobs.py b/src/autoplex/auto/phonons/jobs.py
@@ -398,13 +398,15 @@ def complete_benchmark(  # this function was put here to prevent circular import
     if isinstance(dft_references, list):
         return Response(
             replace=Flow(jobs),
-            output={"bm_output": collect_output, "dft_references": dft_references[ibenchmark_structure]},
-        )
-    else:
-        return Response(
-            replace=Flow(jobs),
-            output={"bm_output": collect_output, "dft_references": dft_references},
+            output={
+                "bm_output": collect_output,
+                "dft_references": dft_references[ibenchmark_structure],
+            },
         )
+    return Response(
+        replace=Flow(jobs),
+        output={"bm_output": collect_output, "dft_references": dft_references},
+    )
 
 
 @job