diff --git a/tests/test_find_map.py b/tests/test_find_map.py index 9f15f2eb..67d22b98 100644 --- a/tests/test_find_map.py +++ b/tests/test_find_map.py @@ -5,10 +5,7 @@ import pytensor.tensor as pt import pytest -from pymc_extras.find_map import scipy_optimize_funcs_from_loss -from pymc_extras.inference.find_map import ( - find_MAP, -) +from pymc_extras.inference.find_map import find_MAP, scipy_optimize_funcs_from_loss pytest.importorskip("jax")