Commit 4b231f3c authored by Theo Steininger's avatar Theo Steininger

Fixed another test for np.int8

parent 45820d90
Pipeline #14849 passed with stage
in 3 minutes and 17 seconds
......@@ -1889,6 +1889,9 @@ class Test_axis(unittest.TestCase):
(a, obj) = generate_data(global_shape, dtype,
distribution_strategy,
strictly_positive=True)
if dtype is np.dtype('int8'):
a[a > 8] = 0
obj[obj > 8] = 0
if function in ['argmin', 'argmax'] and axis is not None:
assert_raises(NotImplementedError, lambda: getattr(obj, function)
(axis=axis))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment