Skip to content

Commit b54d783

Browse files
simplified validation of 'order' arguments to reshape
1 parent b31ddb0 commit b54d783

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

dpctl/tensor/_reshape.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -86,11 +86,9 @@ def reshape(X, newshape, order="C", copy=None):
8686
raise TypeError
8787
if not isinstance(newshape, (list, tuple)):
8888
newshape = (newshape,)
89-
if order in "c":
90-
order = "C"
91-
elif order in "f":
92-
order = "F"
93-
elif order not in ["C", "F"]:
89+
if order in "cfCF":
90+
order = order.upper()
91+
else:
9492
raise ValueError(
9593
f"Keyword 'order' not recognized. Expecting 'C' or 'F', got {order}"
9694
)

0 commit comments

Comments
 (0)