Skip to content

Commit

Permalink
Fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
tehrengruber committed Feb 28, 2025
1 parent 59a1226 commit f832a19
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/gt4py/next/iterator/type_system/type_synthesizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -237,14 +237,14 @@ def concat_where(
result_collection_constructor=lambda el: ts.TupleType(types=list(el)),
)
def deduce_return_type(tb: ts.FieldType | ts.ScalarType, fb: ts.FieldType | ts.ScalarType):
tb_dtype, fb_type = (type_info.extract_dtype(b) for b in [tb, fb])
tb_dtype, fb_dtype = (type_info.extract_dtype(b) for b in [tb, fb])

assert (
tb_dtype == fb_type
), f"Field arguments must be of same dtype, got '{tb_dtype}' != '{fb_type}'."
tb_dtype == fb_dtype
), f"Field arguments must be of same dtype, got '{tb_dtype}' != '{fb_dtype}'."

return_dims = common.promote_dims(domain.dims, type_info.promote(tb, fb).dims)
return_type = ts.FieldType(dims=return_dims, dtype=type_info.promote(t_dtype, f_dtype))
return_type = ts.FieldType(dims=return_dims, dtype=type_info.promote(tb_dtype, fb_dtype))
return return_type

return deduce_return_type(true_field, false_field)
Expand Down

0 comments on commit f832a19

Please sign in to comment.