|
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