def untile_axis(out, axis: int | None):
  if axis is None:
    return out
  shape = list(out.shape)
  shape[axis:axis+2] = [shape[axis] * shape[axis+1]]

ERROR `*` is not supported between `list[@_]` and `list[@_]` [unsupported-operation]
 --> /tmp/foo.py:5:25
  |
5 |   shape[axis:axis+2] = [shape[axis] * shape[axis+1]]
  |                         ^^^^^^^^^^^^^^^^^^^^^^^^^^^
  |
  Argument `list[@_]` is not assignable to parameter `value` with type `SupportsIndex` in function `list.__mul__`
  Protocol `SupportsIndex` requires attribute `__index__`
ERROR `*` is not supported between `list[list[Unknown]]` and `list[list[Unknown]]` [unsupported-operation]
 --> /tmp/foo.py:5:25
  |
5 |   shape[axis:axis+2] = [shape[axis] * shape[axis+1]]
  |                         ^^^^^^^^^^^^^^^^^^^^^^^^^^^
  |
  Argument `list[list[Unknown]]` is not assignable to parameter `value` with type `SupportsIndex` in function `list.__mul__`
  Protocol `SupportsIndex` requires attribute `__index__`
 INFO 2 errors