Skip to content

Commit 3c9d902

Browse files
committed
fixes
1 parent bc4ae6a commit 3c9d902

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

adaptive/learner/learner2D.py

+6-3
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
from adaptive.learner.base_learner import BaseLearner
1616
from adaptive.learner.triangulation import simplex_volume_in_embedding
1717
from adaptive.notebook_integration import ensure_holoviews
18+
from adaptive.types import Bool
1819
from adaptive.utils import (
1920
assign_defaults,
2021
cache_latest,
@@ -530,7 +531,9 @@ def bounds_are_done(self) -> bool:
530531
(p in self.pending_points or p in self._stack) for p in self._bounds_points
531532
)
532533

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]:
534537
"""Get the interpolated data on a grid.
535538
536539
Parameters
@@ -593,7 +596,7 @@ def _data_combined(self) -> tuple[np.ndarray, np.ndarray]:
593596
values_combined = np.vstack([values, values_interp])
594597
return points_combined, values_combined
595598

596-
def ip(self):
599+
def ip(self) -> LinearNDInterpolator:
597600
"""Deprecated, use `self.interpolator(scaled=True)`"""
598601
warnings.warn(
599602
"`learner.ip()` is deprecated, use `learner.interpolator(scaled=True)`."
@@ -643,7 +646,7 @@ def _interpolator_combined(self) -> LinearNDInterpolator:
643646
self._ip_combined = interpolate.LinearNDInterpolator(points, values)
644647
return self._ip_combined
645648

646-
def inside_bounds(self, xy: tuple[float, float]) -> bool | np.bool_:
649+
def inside_bounds(self, xy: tuple[float, float]) -> Bool:
647650
x, y = xy
648651
(xmin, xmax), (ymin, ymax) = self.bounds
649652
return xmin <= x <= xmax and ymin <= y <= ymax

adaptive/learner/learnerND.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
simplex_volume_in_embedding,
2626
)
2727
from adaptive.notebook_integration import ensure_holoviews, ensure_plotly
28+
from adaptive.types import Bool
2829
from adaptive.utils import (
2930
assign_defaults,
3031
cache_latest,
@@ -574,7 +575,7 @@ def _simplex_exists(self, simplex: Simplex) -> bool:
574575
simplex = tuple(sorted(simplex))
575576
return simplex in self.tri.simplices
576577

577-
def inside_bounds(self, point: tuple[float, ...]) -> bool | np.bool_:
578+
def inside_bounds(self, point: tuple[float, ...]) -> Bool:
578579
"""Check whether a point is inside the bounds."""
579580
if self._interior is not None:
580581
return self._interior.find_simplex(point, tol=1e-8) >= 0

0 commit comments

Comments
 (0)