export RNGKey

lenskit · Oct 5, 2024 · b6ebd31 · b6ebd31
1 parent 331b027
commit b6ebd31
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/seedbank/__init__.py b/seedbank/__init__.py
@@ -42,6 +42,7 @@
     "numpy_random_state",
     "cupy_rng",
     "jax_key",
+    "RNGKey",
     "SeedLike",
 ]