diff --git a/symba/base.py b/symba/base.py index d2083e7..5d815bd 100644 --- a/symba/base.py +++ b/symba/base.py @@ -8,22 +8,6 @@ from .core.term import Term as _Term from .hints import SqrtEvaluator as _SqrtEvaluator -Expression = Expression - - -def get_sqrt_evaluator() -> _SqrtEvaluator: - """ - Returns current square root evaluator. - """ - return _context.sqrt_evaluator.get() - - -def set_sqrt_evaluator(evaluator: _SqrtEvaluator) -> None: - """ - Sets square root evaluator. - """ - _context.sqrt_evaluator.set(evaluator) - def sqrt(argument: _Union[_Real, Expression]) -> Expression: """ @@ -43,3 +27,20 @@ def sqrt(argument: _Union[_Real, Expression]) -> Expression: if argument < 0: raise ValueError('Argument should be non-negative.') return _Term.from_components(_One, _to_constant(argument)) + + +Expression = Expression + + +def get_sqrt_evaluator() -> _SqrtEvaluator: + """ + Returns current square root evaluator. + """ + return _context.sqrt_evaluator.get() + + +def set_sqrt_evaluator(evaluator: _SqrtEvaluator) -> None: + """ + Sets square root evaluator. + """ + _context.sqrt_evaluator.set(evaluator)