From 4afba20657467dfe223b0444e59b03068bf76bb1 Mon Sep 17 00:00:00 2001 From: Ricardo Vieira Date: Tue, 1 Aug 2023 10:26:21 +0200 Subject: [PATCH] Add numpy constants to pytensor.tensor scope --- pytensor/tensor/__init__.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pytensor/tensor/__init__.py b/pytensor/tensor/__init__.py index dfe74b5b2f..e5bc217e74 100644 --- a/pytensor/tensor/__init__.py +++ b/pytensor/tensor/__init__.py @@ -148,5 +148,8 @@ def _get_vector_length_Constant(op: Union[Op, Variable], var: Constant) -> int: from pytensor.tensor.type_other import * # noqa from pytensor.tensor.var import TensorConstant, TensorVariable # noqa +# Allow accessing numpy constants from pytensor.tensor +from numpy import e, euler_gamma, inf, infty, nan, newaxis, pi # noqa + __all__ = ["random"] # noqa: F405