Fix black test failure in CI

.github/workflows/ci.yml

@@ -17,10 +17,10 @@ jobs:
       - name:
         uses: actions/checkout@v3
 
-      - name: Set up Python 3.10.5
+      - name: Set up Python 3.10.12
         uses: actions/setup-python@v3
         with:
-          python-version: "3.10.5"
+          python-version: "3.10.12"
 
       - name: Install dependencies
         run: python -m pip install ".[test]"

pyproject.toml

@@ -52,7 +52,7 @@ release = [
 ]
 
 test = [
-  "pytest",
+  "pytest==7.4.4",
   "pytest-black",
   "pytest-cases",
   "pytest-cov",

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] = [