From 7266b2e3a53ef698b95f7c2d2a0bf2f0061f0d73 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Philip=20M=C3=BCller?= <147368808+philip-paul-mueller@users.noreply.github.com> Date: Thu, 22 Aug 2024 09:55:53 +0200 Subject: [PATCH] Added `PatternNode` to `dace.transformation` imports. (#1618) It was very inconvenient, as `PatternTransformation` can be imported from `dace.transformation` but the node has to be imported from `dace.transformation.transformation`. --- dace/transformation/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/dace/transformation/__init__.py b/dace/transformation/__init__.py index 3a4c65efa3..0b27542ca6 100644 --- a/dace/transformation/__init__.py +++ b/dace/transformation/__init__.py @@ -1,3 +1,4 @@ -from .transformation import (PatternTransformation, SingleStateTransformation, MultiStateTransformation, - SubgraphTransformation, ExpandTransformation, experimental_cfg_block_compatible) +from .transformation import (PatternNode, PatternTransformation, SingleStateTransformation, + MultiStateTransformation, SubgraphTransformation, ExpandTransformation, + experimental_cfg_block_compatible, single_level_sdfg_only) from .pass_pipeline import Pass, Pipeline, FixedPointPipeline