Merge branch 'docs_type_update' of https://github.com/nfarabullini/gt4py

into docs_type_update
GridTools · Mar 13, 2024 · bca3ea4 · bca3ea4
2 parents 7157d61 + 81bb432
commit bca3ea4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gt4py/next/embedded/operators.py b/src/gt4py/next/embedded/operators.py
@@ -191,7 +191,7 @@ def _tuple_at(
 ) -> core_defs.Scalar | tuple[core_defs.ScalarT | tuple, ...]:
     @utils.tree_map
     def impl(field: common.Field | core_defs.Scalar) -> core_defs.Scalar:
-        res = field[pos] if common.is_field(field) else field
+        res = field[pos].as_scalar() if common.is_field(field) else field
         assert core_defs.is_scalar_type(res)
         return res