From 65c29fb9df2a98575e5e873de0f91e26eee094a9 Mon Sep 17 00:00:00 2001
From: Thomas Pinder <tompinder@live.co.uk>
Date: Thu, 30 Nov 2023 09:11:28 +0100
Subject: [PATCH] Fix linting

---
 gpjax/__init__.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/gpjax/__init__.py b/gpjax/__init__.py
index 4fadb160..47ba74a3 100644
--- 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",
 ]