Skip to content

Commit 1de8c07

Browse files
committed
fix ordering test
1 parent 9fa3010 commit 1de8c07

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/test_links/test_links.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import keras
12
import numpy as np
23
import pytest
34

@@ -25,6 +26,7 @@ def test_invalid_shape_for_ordered_quantiles_with_specified_axis(
2526

2627

2728
def check_ordering(output, axis):
29+
output = keras.ops.convert_to_numpy(output)
2830
assert np.all(np.diff(output, axis=axis) > 0), f"is not ordered along specified axis: {axis}."
2931
for i in range(output.ndim):
3032
if i != axis % output.ndim:

0 commit comments

Comments
 (0)