Update typing of random key

flferretti · Dec 6, 2023 · b18b930 · b18b930
1 parent 50aea06
commit b18b930
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/jaxsim/high_level/model.py b/src/jaxsim/high_level/model.py
@@ -491,7 +491,7 @@ def joint_positions(self, joint_names: tuple[str, ...] = None) -> jtp.Vector:
 
     @functools.partial(oop.jax_tf.method_ro, static_argnames=["joint_names"])
     def joint_random_positions(
-        self, joint_names: tuple[str, ...] = None, key: jax.random.PRNGKeyArray = None
+        self, joint_names: tuple[str, ...] = None, key: jax.Array = None
     ) -> jtp.Vector:
         """"""