diff --git a/jaxonnxruntime/onnx_ops/reducemax.py b/jaxonnxruntime/onnx_ops/reducemax.py index f602a89..45a206a 100644 --- a/jaxonnxruntime/onnx_ops/reducemax.py +++ b/jaxonnxruntime/onnx_ops/reducemax.py @@ -81,7 +81,7 @@ def onnx_reducemax( keepdims=1, noop_with_empty_axes=0, ): - """The impl for https://github.com/onnx/onnx/blob/v1.12.0/docs/Operators.md#ReduceSum.""" + """The impl for https://github.com/onnx/onnx/blob/v1.12.0/docs/Operators.md#ReduceMax.""" assert len(input_args) == 1 or len(input_args) == 2 data = input_args[0] if axes is None and noop_with_empty_axes > 0: