From 8ff728ba5439ca4a90af3dfd2afa6b7141888ecb Mon Sep 17 00:00:00 2001
From: Crazy-Ryan <30890405+Crazy-Ryan@users.noreply.github.com>
Date: Mon, 9 Jun 2025 15:32:29 +0800
Subject: [PATCH] remove incorrect type hint

---
 python/needle/ops/ops_mathematic.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/needle/ops/ops_mathematic.py b/python/needle/ops/ops_mathematic.py
index f78e024..7521a83 100755
--- a/python/needle/ops/ops_mathematic.py
+++ b/python/needle/ops/ops_mathematic.py
@@ -328,7 +328,7 @@ def __init__(self, axis: int):
         """
         self.axis = axis
 
-    def compute(self, args: TensorTuple) -> Tensor:
+    def compute(self, args):
         ### BEGIN YOUR SOLUTION
         raise NotImplementedError()
         ### END YOUR SOLUTION