Format code with Black

src/wf_psf/data/training_preprocessing.py

@@ -5,6 +5,7 @@
 :Authors: Jennifer Pollack <[email protected]> and Tobias Liaudat <[email protected]>
 
 """
+
 import numpy as np
 import wf_psf.utils.utils as utils
 import tensorflow as tf

src/wf_psf/psf_models/zernikes.py

@@ -5,6 +5,7 @@
 :Author: Tobias Liaudat <[email protected]> and Jennifer Pollack <[email protected]>
 
 """
+
 import numpy as np
 import zernike as zk
 import tensorflow as tf

src/wf_psf/run.py

@@ -5,6 +5,7 @@
 :Author: Jennifer Pollack <[email protected]>
 
 """
+
 import argparse
 from wf_psf.utils.read_config import read_stream
 from wf_psf.utils.io import FileIOHandler

src/wf_psf/tests/conftest.py

@@ -7,6 +7,7 @@
 
 
 """
+
 import pytest
 from wf_psf.utils.read_config import RecursiveNamespace
 from wf_psf.training.train import TrainingParamsHandler

src/wf_psf/tests/metrics_test.py

@@ -5,6 +5,7 @@
 :Author: Jennifer Pollack <[email protected]>
 
 """
+
 import pytest
 from wf_psf.utils.read_config import RecursiveNamespace
 from wf_psf.training import train

src/wf_psf/tests/test_utils/conftest.py

@@ -7,6 +7,7 @@
 
 
 """
+
 import pytest
 import os
 

src/wf_psf/training/train.py

@@ -397,13 +397,13 @@ def train(
 
         # Save optimisation history in the saving dict
         if psf_model.save_optim_history_param:
-            saving_optim_hist[
-                "param_cycle{}".format(current_cycle)
-            ] = hist_param.history
+            saving_optim_hist["param_cycle{}".format(current_cycle)] = (
+                hist_param.history
+            )
         if psf_model.save_optim_history_nonparam:
-            saving_optim_hist[
-                "nonparam_cycle{}".format(current_cycle)
-            ] = hist_non_param.history
+            saving_optim_hist["nonparam_cycle{}".format(current_cycle)] = (
+                hist_non_param.history
+            )
 
     # Save last cycle if no cycles were saved
     if not training_handler.multi_cycle_params.save_all_cycles:

src/wf_psf/utils/configs_handler.py

@@ -6,6 +6,7 @@
 :Authors: Jennifer Pollack <[email protected]>
 
 """
+
 import numpy as np
 from wf_psf.utils.read_config import read_conf
 from wf_psf.data.training_preprocessing import TrainingDataHandler, TestDataHandler
@@ -412,9 +413,9 @@ def call_plot_config_handler_run(self, model_metrics):
         )
 
         # Update metrics_confs dict with latest result
-        plots_config_handler.metrics_confs[
-            self._file_handler.workdir
-        ] = self.metrics_conf
+        plots_config_handler.metrics_confs[self._file_handler.workdir] = (
+            self.metrics_conf
+        )
 
         # Update metric results dict with latest result
         plots_config_handler.list_of_metrics_dict[self._file_handler.workdir] = [