Fix typing

GridTools · Jan 31, 2025 · 6c4ad12 · 6c4ad12
1 parent 9072778
commit 6c4ad12
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/src/gt4py/next/iterator/type_system/type_synthesizer.py b/src/gt4py/next/iterator/type_system/type_synthesizer.py
@@ -141,7 +141,7 @@ def can_deref(it: it_ts.IteratorType | ts.DeferredType) -> ts.ScalarType:
 
 
 @_register_builtin_type_synthesizer
-def if_(pred: ts.ScalarType, true_branch: ts.DataType, false_branch: ts.DataType) -> ts.DataType:
+def if_(pred: ts.ScalarType | ts.DeferredType, true_branch: ts.DataType, false_branch: ts.DataType) -> ts.DataType:
     if isinstance(true_branch, ts.TupleType) and isinstance(false_branch, ts.TupleType):
         return tree_map(
             collection_type=ts.TupleType,

diff --git a/uv.lock b/uv.lock