fix: undo false positive replacement from refactoring

GridTools · Mar 13, 2024 · 57cb2ed · 57cb2ed
1 parent e148c86
commit 57cb2ed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/gt4py/next/ffront/past_to_itir.py b/src/gt4py/next/ffront/past_to_itir.py
@@ -130,10 +130,10 @@ class ProgramLowering(
     --------
     >>> from gt4py.next.ffront.func_to_past import ProgramParser
     >>> from gt4py.next.iterator import ir
-    >>> from gt4py.next import common.Dimension, Field
+    >>> from gt4py.next import Dimension, Field
     >>>
     >>> float64 = float
-    >>> IDim = common.Dimension("IDim")
+    >>> IDim = Dimension("IDim")
     >>>
     >>> def fieldop(inp: Field[[IDim], "float64"]) -> Field[[IDim], "float64"]: ...
     >>> def program(inp: Field[[IDim], "float64"], out: Field[[IDim], "float64"]):