drvs_test import issues fixed

Qazalbash · Dec 24, 2023 · df3f749 · df3f749
1 parent 2d423db
commit df3f749
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/drvs_test.py b/tests/drvs_test.py
@@ -12,11 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import sys
 from itertools import product
 from random import randint
 
 from jax import numpy as jnp
 
+sys.path.append("../jaxampler")
 from jaxampler.rvs.drvs import Bernoulli, Binomial
 
 eps = 1e-3