From a4d9768daedb7a9baa4bc93f73433849dedc6418 Mon Sep 17 00:00:00 2001
From: Robert Knight <robertknight@gmail.com>
Date: Mon, 10 Feb 2025 17:18:42 +0000
Subject: [PATCH] Fix ArgMin op

Fix regression introduced in fee3f09372e3d35abd70005c42c66a425c18ac33 where
ArgMin dispatched to `arg_max` instead of `arg_min`.
---
 src/ops/reduce.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/ops/reduce.rs b/src/ops/reduce.rs
index ec4e6101..65fa7762 100644
--- a/src/ops/reduce.rs
+++ b/src/ops/reduce.rs
@@ -168,7 +168,7 @@ impl Operator for ArgMin {
 
     fn run(&self, pool: &TensorPool, inputs: InputList) -> Result<OutputList, OpError> {
         let input = inputs.require(0)?;
-        dispatch_single_axis_reduce_op!(pool, input, arg_max, self.axis, self.keep_dims)
+        dispatch_single_axis_reduce_op!(pool, input, arg_min, self.axis, self.keep_dims)
     }
 }