From 231a97766195ca313d6bded18a99fee8ab382c86 Mon Sep 17 00:00:00 2001 From: Ricardo Vieira <28983449+ricardoV94@users.noreply.github.com> Date: Mon, 9 Dec 2024 15:08:40 +0100 Subject: [PATCH] Make OpFromGraph available at root scope (#1113) --- pytensor/__init__.py | 1 + 1 file changed, 1 insertion(+) 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