Fix linting

JaxGaussianProcesses · Nov 30, 2023 · 65c29fb · 65c29fb
1 parent ce8f1cd
commit 65c29fb
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/gpjax/__init__.py b/gpjax/__init__.py
@@ -23,10 +23,13 @@
     objectives,
     variational_families,
 )
+from gpjax.base import (
+    Module,
+    param_field,
+)
 from gpjax.citation import cite
 from gpjax.dataset import Dataset
 from gpjax.fit import fit
-from gpjax.base import Module, param_field
 
 __license__ = "MIT"
 __description__ = "Didactic Gaussian processes in JAX"
@@ -48,5 +51,5 @@
     "cite",
     "fit",
     "Module",
-    "param_field"
+    "param_field",
 ]