Apply suggestions from code review

Co-authored-by: Aditya Goel <48102515+adityagoel4512@users.noreply.github.com>
Quantco · Jan 10, 2025 · c70ff40 · c70ff40
1 parent 0563eee
commit c70ff40
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ndonnx/_funcs.py b/ndonnx/_funcs.py
@@ -550,7 +550,7 @@ def matrix_transpose(x):
     return _unary(x.dtype._ops.matrix_transpose, x)
 
 
-def tensordot(x, y, axes=2):
+def tensordot(x, y, /, *, axes=2):
     if (out := x.dtype._ops.tensordot(x, y, axes)) is not NotImplemented:
         return out
     if (out := y.dtype._ops.tensordot(x, y, axes)) is not NotImplemented:
@@ -640,7 +640,7 @@ def permute_dims(x, axes):
     )
 
 
-def repeat(x, repeats, axis=None):
+def repeat(x, repeats, /, *, axis=None):
     if (out := x.dtype._ops.repeat(x, repeats, axis)) is not NotImplemented:
         return out
     raise UnsupportedOperationError(f"Unsupported operand type for repeat: '{x.dtype}'")