zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-02 05:14:40 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Resolved ambiguity in mlx::core::take_along_axis (#1822)

Browse Source 
* Resolved ambiguity in mlx::core::take_along_axis

Detected by GCC 10 on riscv64-linux-gnu.

* Formatted

* Removed superfluous parentheses in random_tests.cpp
This commit is contained in:
Jesper Stemann Andersen
2025-02-04 15:06:17 +01:00
committed by GitHub
parent 1156c84e86
commit f6c0499b8d
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/random_tests.cpp
View File

@@ -556,7 +556,7 @@ TEST_CASE("test random bernoulli") {
p = array({0.1, 0.2, 0.3});
// Ask for the wrong shape => throws
CHECK_THROWS_AS(random::bernoulli(p, Shape({2})), std::invalid_argument);
CHECK_THROWS_AS(random::bernoulli(p, Shape{2}), std::invalid_argument);
// Check wrong key type or shape
auto key = array({0, 0}, {1, 2});