|
15 | 15 | from adaptive.learner.base_learner import BaseLearner
|
16 | 16 | from adaptive.learner.triangulation import simplex_volume_in_embedding
|
17 | 17 | from adaptive.notebook_integration import ensure_holoviews
|
| 18 | +from adaptive.types import Bool |
18 | 19 | from adaptive.utils import (
|
19 | 20 | assign_defaults,
|
20 | 21 | cache_latest,
|
@@ -530,7 +531,9 @@ def bounds_are_done(self) -> bool:
|
530 | 531 | (p in self.pending_points or p in self._stack) for p in self._bounds_points
|
531 | 532 | )
|
532 | 533 |
|
533 |
| - def interpolated_on_grid(self, n=None): |
| 534 | + def interpolated_on_grid( |
| 535 | + self, n: int = None |
| 536 | + ) -> tuple[np.ndarray, np.ndarray, np.ndarray]: |
534 | 537 | """Get the interpolated data on a grid.
|
535 | 538 |
|
536 | 539 | Parameters
|
@@ -593,7 +596,7 @@ def _data_combined(self) -> tuple[np.ndarray, np.ndarray]:
|
593 | 596 | values_combined = np.vstack([values, values_interp])
|
594 | 597 | return points_combined, values_combined
|
595 | 598 |
|
596 |
| - def ip(self): |
| 599 | + def ip(self) -> LinearNDInterpolator: |
597 | 600 | """Deprecated, use `self.interpolator(scaled=True)`"""
|
598 | 601 | warnings.warn(
|
599 | 602 | "`learner.ip()` is deprecated, use `learner.interpolator(scaled=True)`."
|
@@ -643,7 +646,7 @@ def _interpolator_combined(self) -> LinearNDInterpolator:
|
643 | 646 | self._ip_combined = interpolate.LinearNDInterpolator(points, values)
|
644 | 647 | return self._ip_combined
|
645 | 648 |
|
646 |
| - def inside_bounds(self, xy: tuple[float, float]) -> bool | np.bool_: |
| 649 | + def inside_bounds(self, xy: tuple[float, float]) -> Bool: |
647 | 650 | x, y = xy
|
648 | 651 | (xmin, xmax), (ymin, ymax) = self.bounds
|
649 | 652 | return xmin <= x <= xmax and ymin <= y <= ymax
|
|
0 commit comments