ianpan's picture
Fix np.argmax arg
d45106e