diff --git a/pytensor/__init__.py b/pytensor/__init__.py index dd6117c527..24942b6b2c 100644 --- a/pytensor/__init__.py +++ b/pytensor/__init__.py @@ -164,6 +164,7 @@ def get_underlying_scalar_constant(v): from pytensor.scan import checkpoints from pytensor.scan.basic import scan from pytensor.scan.views import foldl, foldr, map, reduce +from pytensor.compile.builders import OpFromGraph # isort: on