update for cosmology.api updates (#9)

* update for cosmology.api updates * tests Signed-off-by: Nathaniel Starkman <nstarman@users.noreply.github.com>
cosmology-api · May 14, 2023 · 7853add · 7853add
1 parent 7caacf4
commit 7853add
Show file tree

Hide file tree

Showing 7 changed files with 356 additions and 138 deletions.
diff --git a/src/cosmology/compat/classy/_core.py b/src/cosmology/compat/classy/_core.py
@@ -18,12 +18,12 @@
     from typing_extensions import TypeAlias
 
 
-NDFloating: TypeAlias = NDArray[floating[Any]]
-InputT: TypeAlias = Union[NDFloating, float]
+Array: TypeAlias = NDArray[floating[Any]]
+InputT: TypeAlias = Union[Array, float]
 
 
 @dataclass(frozen=True)
-class CosmologyWrapper(CosmologyWrapperAPI[NDFloating, InputT]):
+class CosmologyWrapper(CosmologyWrapperAPI[Array, InputT]):
     """The Cosmology API wrapper for :mod:`classy`."""
 
     cosmo: classy.Class