def METHOD_NAME(self):
    if isinstance(self.sample_normalized_shape, numbers.Integral):
        # mypy error: incompatible types in assignment
        sample_normalized_shape = (self.sample_normalized_shape,)  # type: ignore[assignment]
    else:
        sample_normalized_shape = self.sample_normalized_shape
    return tuple(sample_normalized_shape)  # type: ignore[arg-type]