Skip to content

Commit fb47422

Browse files
Break MaxandArgmax to seperate TensorMax and Argmax
1 parent 151f954 commit fb47422

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/tensor/test_math.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -763,7 +763,7 @@ def setup_method(self):
763763
MaxAndArgmax.debug = 0
764764

765765
def test_basic(self):
766-
n = as_tensor_variable([5.0])
766+
n = as_tensor_variable(5.0)
767767
v, i = eval_outputs(max_and_argmax(n))
768768
assert v == 5.0
769769
assert i == 0

0 commit comments

Comments
 (0)