From 13cd6e4cd328aa3f47a568c83648ebd1f8ba362b Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 18 Oct 2024 22:31:21 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- .../laser_ion/analysis_time_avg_diags.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/Examples/Physics_applications/laser_ion/analysis_time_avg_diags.py b/Examples/Physics_applications/laser_ion/analysis_time_avg_diags.py index 3a24af91d0d..74a4687cec4 100755 --- a/Examples/Physics_applications/laser_ion/analysis_time_avg_diags.py +++ b/Examples/Physics_applications/laser_ion/analysis_time_avg_diags.py @@ -1,11 +1,14 @@ #!/usr/bin/env python3 import sys + import numpy as np import openpmd_api as io -def load_field_from_iteration(series, iteration : int, field : str, coord : str = None) -> np.ndarray: +def load_field_from_iteration( + series, iteration: int, field: str, coord: str = None +) -> np.ndarray: """Load iteration of field data from file.""" it = series.iterations[iteration] @@ -16,14 +19,15 @@ def load_field_from_iteration(series, iteration : int, field : str, coord : str elif coord in [item[0] for item in list(field_obj.items())]: field_data = field_obj[coord].load_chunk() else: - raise Exception(f"Specified coordinate: f{coord} is not available for field: f{field}.") + raise Exception( + f"Specified coordinate: f{coord} is not available for field: f{field}." + ) series.flush() return field_data def main(): - field = "E" coord = "z" avg_period_steps = 5 @@ -43,7 +47,7 @@ def main(): data_inst = np.zeros(shape) - for i in np.arange(avg_output_step-avg_period_steps+1,avg_output_step+1): + for i in np.arange(avg_output_step - avg_period_steps + 1, avg_output_step + 1): data_inst += load_field_from_iteration(si, i, field, coord) data_inst = data_inst / avg_period_steps @@ -59,4 +63,4 @@ def main(): if __name__ == "__main__": - main() \ No newline at end of file + main()