enforce strict abstract-concrete (#71)

Signed-off-by: nstarman <nstarman@users.noreply.github.com>
GalacticDynamics · Jan 19, 2024 · c6480a9 · c6480a9
1 parent 805666f
commit c6480a9
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/src/galax/dynamics/_core.py b/src/galax/dynamics/_core.py
@@ -18,7 +18,7 @@
     from galax.potential._potential.base import AbstractPotentialBase
 
 
-class AbstractPhaseSpacePositionBase(eqx.Module):  # type: ignore[misc]
+class AbstractPhaseSpacePositionBase(eqx.Module, strict=True):  # type: ignore[call-arg, misc]
     """Abstract Base Class of Phase-Space Positions.
 
     Todo:

diff --git a/src/galax/dynamics/mockstream/_df/base.py b/src/galax/dynamics/mockstream/_df/base.py
@@ -20,7 +20,7 @@
 Carry: TypeAlias = tuple[IntLike, Vec3, Vec3, Vec3, Vec3]
 
 
-class AbstractStreamDF(eqx.Module):  # type: ignore[misc]
+class AbstractStreamDF(eqx.Module, strict=True):  # type: ignore[call-arg, misc]
     lead: bool = eqx.field(default=True, static=True)
     trail: bool = eqx.field(default=True, static=True)
 

diff --git a/src/galax/integrate/_base.py b/src/galax/integrate/_base.py
@@ -31,7 +31,7 @@ def __call__(self, t: FloatScalar, qp: Vec6, args: tuple[Any, ...]) -> Vec6:
         ...
 
 
-class AbstractIntegrator(eqx.Module):  # type: ignore[misc]
+class AbstractIntegrator(eqx.Module, strict=True):  # type: ignore[call-arg, misc]
     """Integrator Class.
 
     The integrators are classes that are used to integrate the equations of

diff --git a/src/galax/potential/_potential/base.py b/src/galax/potential/_potential/base.py
@@ -37,7 +37,7 @@
 default_integrator = DiffraxIntegrator()
 
 
-class AbstractPotentialBase(eqx.Module, metaclass=ModuleMeta):  # type: ignore[misc]
+class AbstractPotentialBase(eqx.Module, metaclass=ModuleMeta, strict=True):  # type: ignore[misc]
     """Potential Class."""
 
     units: eqx.AbstractVar[UnitSystem]

diff --git a/src/galax/potential/_potential/core.py b/src/galax/potential/_potential/core.py
@@ -13,7 +13,7 @@
 from .utils import converter_to_usys
 
 
-class AbstractPotential(AbstractPotentialBase):
+class AbstractPotential(AbstractPotentialBase, strict=True):
     _: KW_ONLY
     units: UnitSystem = eqx.field(converter=converter_to_usys, static=True)
     _G: float = eqx.field(init=False, static=True, repr=False, converter=float)

diff --git a/src/galax/potential/_potential/param/core.py b/src/galax/potential/_potential/param/core.py
@@ -20,7 +20,7 @@
 from galax.utils.dataclasses import converter_float_array
 
 
-class AbstractParameter(eqx.Module):  # type: ignore[misc]
+class AbstractParameter(eqx.Module, strict=True):  # type: ignore[call-arg, misc]
     """Abstract Base Class for Parameters on a Potential.
 
     Parameters are time-dependent quantities that are used to define a