run black

zincware · May 16, 2024 · e422567 · e422567
1 parent afe6d1a
commit e422567
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/znnl/training_recording/papyrus_jax_recording.py b/znnl/training_recording/papyrus_jax_recording.py
@@ -195,7 +195,7 @@ def _compute_neural_state(self, model: JaxModel):
         Parameters
         ----------
         model : JaxModel
-                The neural network module containing the parameters to use for 
+                The neural network module containing the parameters to use for
                 recording.
 
         Returns
@@ -226,7 +226,7 @@ def record(self, epoch: int, model: JaxModel, **kwargs):
         epoch : int
                 The epoch of the training process.
         model : JaxModel
-                The neural network module containing the parameters to use for 
+                The neural network module containing the parameters to use for
                 recording.
         kwargs : Any
                 Additional keyword arguments that are directly added to the neural