diff --git a/src/pystencils/backend/functions.py b/src/pystencils/backend/functions.py index ea0d6cb9d188061f305b19760001860e2582e3d1..73634539505e789963b62072f14a28c0586a65a5 100644 --- a/src/pystencils/backend/functions.py +++ b/src/pystencils/backend/functions.py @@ -94,7 +94,7 @@ class MathFunctions(Enum): self.num_args = num_args -class NumericLimitsFunctions(MathFunctions): +class NumericLimitsFunctions(Enum): """Numerical limits functions supported by the backend. Each platform has to materialize these functions to a concrete implementation. @@ -109,12 +109,12 @@ class PsMathFunction(PsFunction): __match_args__ = ("func",) - def __init__(self, func: MathFunctions) -> None: + def __init__(self, func: MathFunctions | NumericLimitsFunctions) -> None: super().__init__(func.function_name, func.num_args) self._func = func @property - def func(self) -> MathFunctions: + def func(self) -> MathFunctions | NumericLimitsFunctions: return self._func def __str__(self) -> str: