Fixes ArgMax deprecation warnings in test (#11551)
The recent commit cbe1ef05 marks the `dimension` argument of argmax and argmin as deprecated, which are used in the corresponding tests and so this brings up warnings when the tests are run. The tests should be using the `axis` argument instead.
Loading
Please sign in to comment