-
Notifications
You must be signed in to change notification settings - Fork 47
Closed
Description
For example, in assert_dtype, the in_dtype is before out_dtype
array-api-tests/array_api_tests/pytest_helpers.py
Lines 83 to 90 in 511929c
| def assert_dtype( | |
| func_name: str, | |
| in_dtype: Union[DataType, Sequence[DataType]], | |
| out_dtype: DataType, | |
| expected: Optional[DataType] = None, | |
| *, | |
| repr_name: str = "out.dtype", | |
| ): |
assert_keepdimable_shape the in_shape is after out_shape array-api-tests/array_api_tests/pytest_helpers.py
Lines 181 to 189 in 511929c
| 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.
Metadata
Metadata
Assignees
Labels
No labels