def specialize_IndexMap(fn, n_indices):
  idx_type = make_tuple_type( (Int64,) + n_indices) if n_indices > 1 else Int64
  typed_fn = specialize(fn, (idx_type,))
  result_type = array_type.increase_rank(typed_fn.return_type, n_indices)
  return result_type, typed_fn
dataset/ETHPy150Open iskandr/parakeet/parakeet/type_inference/type_inference.py/specialize_IndexMap/operator `*` -> `+`