Updated the make_array() function.

Before the `order` argument was a `Literal` but this caused more truble now it is a string.

tests/integration_tests/primitive_translators/test_primitive_reshape.py

@@ -25,8 +25,7 @@ def _test_impl_reshaping(
     src_shape: Sequence[int], dst_shape: Sequence[int], order: str = "C"
 ) -> None:
     """Performs a reshaping from `src_shape` to `dst_shape`."""
-    a = testutil.make_array(src_shape)
-    a = np.array(a, order=order)  # type: ignore[call-overload]  # MyPy wants a literal as order.
+    a = testutil.make_array(src_shape, order=order)
 
     def testee(a: np.ndarray) -> jax.Array:
         return jnp.reshape(a, dst_shape)

tests/util.py

@@ -10,7 +10,7 @@
 from __future__ import annotations
 
 from collections.abc import Mapping, Sequence
-from typing import Any, Literal
+from typing import Any
 
 import numpy as np
 
@@ -23,7 +23,7 @@
 def make_array(
     shape: Sequence[int] | int,
     dtype: type = np.float64,
-    order: Literal[None, "K", "A", "C", "F"] = "C",
+    order: str = "C",
     low: Any = None,
     high: Any = None,
 ) -> np.ndarray:
@@ -69,7 +69,7 @@ def make_array(
             res = low + (high - low) * res
         assert (low is None) == (high is None)
 
-    return np.array(res, order=order, dtype=dtype)
+    return np.array(res, order=order, dtype=dtype)  # type: ignore[call-overload]  # Because we use `str` as `order`.
 
 
 def set_active_primitive_translators_to(