fixes

Author: basnijholt

adaptive/learner/base_learner.py

@@ -124,7 +124,7 @@ def remove_unfinished(self):
         """Remove uncomputed data from the learner."""
 
     @abc.abstractmethod
-    def loss(self, real: bool = True):
+    def loss(self, real: bool = True) -> float:
         """Return the loss for the current state of the learner.
 
         Parameters

adaptive/learner/triangulation.py

@@ -28,10 +28,17 @@
 from numpy.linalg import det as ndet
 from numpy.linalg import matrix_rank, norm, slogdet, solve
 
-SimplexPoints = Union[List[Tuple[float, ...]], ndarray]
-Simplex = Union[Sequence[numbers.Integral], ndarray]
-Point = Union[Tuple[float, ...], ndarray]
-Points = Union[Sequence[Tuple[float, ...]], ndarray]
+try:
+    from typing import TypeAlias
+except ImportError:
+    # Remove this when we drop support for Python 3.9
+    from typing_extensions import TypeAlias
+
+
+SimplexPoints: TypeAlias = Union[List[Tuple[float, ...]], ndarray]
+Simplex: TypeAlias = Union[Sequence[numbers.Integral], ndarray]
+Point: TypeAlias = Union[Tuple[float, ...], ndarray]
+Points: TypeAlias = Union[Sequence[Tuple[float, ...]], ndarray]
 
 
 def fast_norm(v: tuple[float, ...] | ndarray) -> float: