Lints according to Ruff

JaxGaussianProcesses · Jul 1, 2024 · 1f67bfb · 1f67bfb
1 parent b455c9a
commit 1f67bfb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/gpjax/decision_making/utility_functions/__init__.py b/gpjax/decision_making/utility_functions/__init__.py
@@ -18,10 +18,10 @@
     SinglePointUtilityFunction,
     UtilityFunction,
 )
-from gpjax.decision_making.utility_functions.thompson_sampling import ThompsonSampling
 from gpjax.decision_making.utility_functions.probability_of_improvement import (
     ProbabilityOfImprovement,
 )
+from gpjax.decision_making.utility_functions.thompson_sampling import ThompsonSampling
 
 __all__ = [
     "UtilityFunction",

diff --git a/tests/test_decision_making/test_utility_functions/test_probability_of_improvement.py b/tests/test_decision_making/test_utility_functions/test_probability_of_improvement.py
@@ -16,14 +16,17 @@
 
 config.update("jax_enable_x64", True)
 
-import jax.random as jr
 import jax.numpy as jnp
+import jax.random as jr
 
 from gpjax.decision_making.test_functions.continuous_functions import Forrester
 from gpjax.decision_making.utility_functions.probability_of_improvement import (
     ProbabilityOfImprovement,
 )
-from gpjax.decision_making.utils import OBJECTIVE, gaussian_cdf
+from gpjax.decision_making.utils import (
+    OBJECTIVE,
+    gaussian_cdf,
+)
 from tests.test_decision_making.utils import generate_dummy_conjugate_posterior