fixed imports

HMUNACHI · Mar 14, 2024 · 7246805 · 7246805
1 parent 2af3047
commit 7246805
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_random.py b/tests/test_random.py
@@ -2,7 +2,7 @@
 import jax.numpy as jnp
 from nanodl import (
     time_rng_key, uniform, normal, bernoulli, categorical, randint,
-    permutation, gumbel, choice, binomial, bits, exponential,
+    permutation, gumbel, choice, bits, exponential,
     triangular, truncated_normal, poisson, geometric, gamma,
     chisquare
 )