Inconsistent arguments in some helper functions
For example, in assert_dtype, the in_dtype is before out_dtype
| def assert_dtype( | |
| func_name: str, | |
| in_dtype: Union[DataType, Sequence[DataType]], | |
| out_dtype: DataType, | |
| expected: Optional[DataType] = None, | |
| *, | |
| repr_name: str = "out.dtype", | |
| ): |
in assert_keepdimable_shape the in_shape is after out_shape
| def assert_keepdimable_shape( | |
| func_name: str, | |
| out_shape: Shape, | |
| in_shape: Shape, | |
| axes: Tuple[int, ...], | |
| keepdims: bool, | |
| /, | |
| **kw, | |
| ): |
By the way, it would be useful to have small docstrings for the various helper functions.