Run linters

Author: thomaspinder

examples/heteroscedastic_inference.py

@@ -44,8 +44,8 @@
 from jax import config
 import jax.numpy as jnp
 import jax.random as jr
-import matplotlib.pyplot as plt
 import matplotlib as mpl
+import matplotlib.pyplot as plt
 import optax as ox
 
 from examples.utils import use_mpl_style
@@ -55,7 +55,6 @@
     LogNormalTransform,
     SoftplusTransform,
 )
-from gpjax.objectives import heteroscedastic_elbo
 from gpjax.variational_families import (
     HeteroscedasticVariationalFamily,
     VariationalGaussianInit,

gpjax/citation.py

@@ -23,7 +23,6 @@
     Matern32,
     Matern52,
 )
-
 from gpjax.likelihoods import HeteroscedasticGaussian
 
 CitationType = Union[None, str, Dict[str, str]]

tests/conftest.py

@@ -1,5 +1,5 @@
-from jax import config
 from hypothesis import settings
+from jax import config
 from jaxtyping import install_import_hook
 
 config.update("jax_enable_x64", True)

tests/integration_tests.py

@@ -28,6 +28,7 @@
 )
 import jax.numpy as jnp  # noqa: F401
 import jupytext
+
 import gpjax
 
 # %%

tests/test_citations.py

@@ -23,8 +23,6 @@
     Matern32,
     Matern52,
 )
-
-
 from gpjax.likelihoods import HeteroscedasticGaussian
 from gpjax.mean_functions import Zero
 

tests/test_heteroscedastic.py

@@ -13,12 +13,15 @@
 # limitations under the License.
 # ==============================================================================
 from flax import nnx
+from hypothesis import (
+    given,
+    strategies as st,
+)
 import jax
 from jax import config
 import jax.numpy as jnp
 import jax.random as jr
 import pytest
-from hypothesis import given, settings, strategies as st
 
 import gpjax as gpx
 from gpjax.dataset import Dataset