Skip to content

Commit 0321629

Browse files
committed
Validation of integer indexing modes fixed
- No longer permits passing 2 as a flag (leftover from changes to clip and wrap modes)
1 parent 9ca3b6b commit 0321629

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

dpctl/tensor/libtensor/source/integer_advanced_indexing.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -253,8 +253,8 @@ usm_ndarray_take(dpctl::tensor::usm_ndarray src,
253253
throw py::value_error("Axis cannot be negative.");
254254
}
255255

256-
if (mode != 0 && mode != 1 && mode != 2) {
257-
throw py::value_error("Mode must be 0, 1, or 2.");
256+
if (mode != 0 && mode != 1) {
257+
throw py::value_error("Mode must be 0 or 1.");
258258
}
259259

260260
const dpctl::tensor::usm_ndarray ind_rep = ind[0];
@@ -564,8 +564,8 @@ usm_ndarray_put(dpctl::tensor::usm_ndarray dst,
564564
throw py::value_error("Axis cannot be negative.");
565565
}
566566

567-
if (mode != 0 && mode != 1 && mode != 2) {
568-
throw py::value_error("Mode must be 0, 1, or 2.");
567+
if (mode != 0 && mode != 1) {
568+
throw py::value_error("Mode must be 0 or 1.");
569569
}
570570

571571
if (!dst.is_writable()) {

0 commit comments

Comments
 (0)