Skip to content

Conversation

@ev-br
Copy link
Member

@ev-br ev-br commented Nov 22, 2025

@ev-br ev-br force-pushed the take_along_axis_torch branch from d2c238a to 2784516 Compare November 22, 2025 16:19
@ev-br ev-br changed the title ENH: test torch.test_along_axis with indices < 0 ENH: test take and take_along_axis with indices < 0 Nov 22, 2025
@ev-br
Copy link
Member Author

ev-br commented Dec 2, 2025

$ ARRAY_API_TESTS_SKIP_DTYPES=uint16,uint32,uint64 ARRAY_API_TESTS_MODULE=array_api_compat.torch pytest array_api_tests/test_indexing_functions.py::test_take_along_axis --max-examples 50000 

and

$ ARRAY_API_TESTS_SKIP_DTYPES=uint16,uint32,uint64 ARRAY_API_TESTS_MODULE=array_api_compat.torch pytest array_api_tests/test_indexing_functions.py::test_take --max-examples 50000 

both pass with 50_000 examples locally: pass with data-apis/array-api-compat#361 and fail with array-api-compat/main. Let's give it a spin.

@ev-br ev-br merged commit b7ee3d2 into data-apis:master Dec 2, 2025
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant