Skip to content

Commit

Permalink
fix: Change deprecated jax type jax.random.KeyArray to jax.Array (#175)
Browse files Browse the repository at this point in the history
  • Loading branch information
LisaCoiffard authored Feb 13, 2024
1 parent a721221 commit eb2cf31
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion qdax/types.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,5 +45,5 @@ def __init__(self) -> None:
Mask: TypeAlias = jnp.ndarray

# Others
RNGKey: TypeAlias = jax.random.KeyArray
RNGKey: TypeAlias = jax.Array
Metrics: TypeAlias = Dict[str, jnp.ndarray]

0 comments on commit eb2cf31

Please sign in to comment.