Update orttraining/orttraining/python/training/api/module.py

Co-authored-by: pengwa <[email protected]>
microsoft · Jul 14, 2023 · 857defc · 857defc
1 parent 4060226
commit 857defc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/orttraining/orttraining/python/training/api/module.py b/orttraining/orttraining/python/training/api/module.py
@@ -66,7 +66,7 @@ def __init__(
         )
         self._state = state
 
-    def __call__(self, *user_inputs) -> tuple[np.ndarray] | np.ndarray | tuple[OrtValue] | OrtValue:
+    def __call__(self, *user_inputs) -> tuple[np.ndarray, ...] | np.ndarray | tuple[OrtValue, ...] | OrtValue:
         """Invokes either the training or the evaluation step of the model.
 
         Args: