fix clever bug

ins-amu · Feb 16, 2024 · 601d62e · 601d62e
1 parent 2e178cc
commit 601d62e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vbjax/util.py b/vbjax/util.py
@@ -8,7 +8,7 @@ def to_np(x: jax.numpy.ndarray) -> numpy.ndarray:
         return numpy.array(x)
     return numpy.from_dlpack(x)
 
-def to_jax(*x: numpy.ndarray):
+def to_jax(x: numpy.ndarray):
     "Move NumPy array to JAX via DLPack."
     x_dlp = x.__dlpack__()
     x_jax: jax.numpy.ndarray = jax.dlpack.from_dlpack(x_dlp)