Import fixes

Author: basnijholt

adaptive/__init__.py

@@ -1,6 +1,5 @@
 from contextlib import suppress
 
-from adaptive import learner, runner, utils
 from adaptive._version import __version__
 from adaptive.learner import (
     AverageLearner,
@@ -22,6 +21,8 @@
 )
 from adaptive.runner import AsyncRunner, BlockingRunner, Runner
 
+from adaptive import learner, runner, utils  # isort:skip
+
 __all__ = [
     "learner",
     "runner",

adaptive/runner.py

@@ -14,15 +14,13 @@
 import warnings
 from contextlib import suppress
 from datetime import datetime, timedelta
-from typing import TYPE_CHECKING, Any, Callable
+from typing import Any, Callable
 
 import loky
 
+from adaptive import BalancingLearner, BaseLearner, IntegratorLearner, SequenceLearner
 from adaptive.notebook_integration import in_ipynb, live_info, live_plot
 
-if TYPE_CHECKING:
-    from adaptive import BaseLearner
-
 try:
     import ipyparallel
 
@@ -918,8 +916,6 @@ def auto_goal(
     -------
     Callable[[adaptive.BaseLearner], bool]
     """
-    from adaptive import BalancingLearner, IntegratorLearner, SequenceLearner
-
     if callable(goal):
         return goal
     if isinstance(goal, float):