diff --git a/cca_zoo/classical/_iterative/_altmaxvar.py b/cca_zoo/classical/_iterative/_altmaxvar.py
index a8235c13..bbf66b4d 100644
--- a/cca_zoo/classical/_iterative/_altmaxvar.py
+++ b/cca_zoo/classical/_iterative/_altmaxvar.py
@@ -36,7 +36,7 @@ def __init__(
             convergence_checking=convergence_checking,
             track=track,
             verbose=verbose,
-            trainer_kwargs={"accelerator": "cpu"}
+            trainer_kwargs={"accelerator": "cpu"},
         )
         self.tau = tau
         self.proximal = proximal
diff --git a/cca_zoo/classical/_iterative/_elasticnet.py b/cca_zoo/classical/_iterative/_elasticnet.py
index 8983f91f..1c200996 100644
--- a/cca_zoo/classical/_iterative/_elasticnet.py
+++ b/cca_zoo/classical/_iterative/_elasticnet.py
@@ -101,7 +101,7 @@ def __init__(
             patience=0,
             track=track,
             verbose=verbose,
-            trainer_kwargs={"accelerator": "cpu"}
+            trainer_kwargs={"accelerator": "cpu"},
         )
 
     def _check_params(self):
diff --git a/cca_zoo/classical/_iterative/_pls_als.py b/cca_zoo/classical/_iterative/_pls_als.py
index 57b76016..6f4e1e58 100644
--- a/cca_zoo/classical/_iterative/_pls_als.py
+++ b/cca_zoo/classical/_iterative/_pls_als.py
@@ -74,7 +74,7 @@ def __init__(
             learning_rate=learning_rate,
             initialization=initialization,
             callbacks=callbacks,
-            trainer_kwargs={"accelerator": "cpu"}
+            trainer_kwargs={"accelerator": "cpu"},
         )
 
     def _get_module(self, weights=None, k=None):
diff --git a/cca_zoo/classical/_iterative/_pmd.py b/cca_zoo/classical/_iterative/_pmd.py
index 352d1adc..6a896e1d 100644
--- a/cca_zoo/classical/_iterative/_pmd.py
+++ b/cca_zoo/classical/_iterative/_pmd.py
@@ -88,7 +88,7 @@ def __init__(
             patience=0,
             track=track,
             verbose=verbose,
-            trainer_kwargs={"accelerator": "cpu"}
+            trainer_kwargs={"accelerator": "cpu"},
         )
         self.tau = tau
         self.positive = positive
diff --git a/cca_zoo/classical/_iterative/_scca_parkhomenko.py b/cca_zoo/classical/_iterative/_scca_parkhomenko.py
index 6f1b0adb..c49ca8a2 100644
--- a/cca_zoo/classical/_iterative/_scca_parkhomenko.py
+++ b/cca_zoo/classical/_iterative/_scca_parkhomenko.py
@@ -69,7 +69,7 @@ def __init__(
             patience=patience,
             track=track,
             verbose=verbose,
-            trainer_kwargs={"accelerator": "cpu"}
+            trainer_kwargs={"accelerator": "cpu"},
         )
 
     def _check_params(self):
diff --git a/cca_zoo/classical/_iterative/_scca_span.py b/cca_zoo/classical/_iterative/_scca_span.py
index 53168900..6d10c08f 100644
--- a/cca_zoo/classical/_iterative/_scca_span.py
+++ b/cca_zoo/classical/_iterative/_scca_span.py
@@ -61,7 +61,7 @@ def __init__(
             random_state=random_state,
             deflation=deflation,
             verbose=verbose,
-            trainer_kwargs={"accelerator": "cpu"}
+            trainer_kwargs={"accelerator": "cpu"},
         )
         self.tau = tau
         self.regularisation = regularisation