diff --git a/src/pyepal/pal/pal_neural_tangent.py b/src/pyepal/pal/pal_neural_tangent.py
index 61ed564..455b3be 100644
--- a/src/pyepal/pal/pal_neural_tangent.py
+++ b/src/pyepal/pal/pal_neural_tangent.py
@@ -37,6 +37,7 @@
 
 __all__ = ["PALNT", "NTModel"]
 
+
 # We move those functions out of the class so that we can parallelize them
 def _set_one_infinite_width_model(  # pylint:disable=too-many-arguments
     i: int,