fix: add 'impl' kwarg to jax PRNGKey

ivy-llc · May 29, 2024 · 9a5a578 · 9a5a578
1 parent 827459d
commit 9a5a578
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ivy/functional/frontends/jax/random.py b/ivy/functional/frontends/jax/random.py
@@ -30,7 +30,7 @@ def _remove_axis(shape, axis):
 
 
 @to_ivy_arrays_and_back
-def PRNGKey(seed):
+def PRNGKey(seed, *, impl=None):
     return ivy.array([0, seed % 4294967295 - (seed // 4294967295)], dtype=ivy.int64)