address review comments

GridTools · Dec 3, 2024 · 7fb18ae · 7fb18ae
1 parent f220b34
commit 7fb18ae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/gt4py/next/iterator/type_system/inference.py b/src/gt4py/next/iterator/type_system/inference.py
@@ -272,8 +272,8 @@ def _get_dimensions(obj: Any):
         if isinstance(obj, common.Dimension):
             yield obj
         elif isinstance(obj, ts.TypeSpec):
-            for field in obj.__datamodel_fields__.values():
-                yield from _get_dimensions(getattr(obj, field.name))
+            for field in obj.__datamodel_fields__.keys():
+                yield from _get_dimensions(getattr(obj, field))
         elif isinstance(obj, collections.abc.Mapping):
             for el in obj.values():
                 yield from _get_dimensions(el)