fix test import

EleutherAI · Mar 8, 2023 · 801440c · 801440c
1 parent 1cafc51
commit 801440c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_math.py b/tests/test_math.py
@@ -1,4 +1,4 @@
-from elk.math import stochastic_round_constrained
+from elk.math_util import stochastic_round_constrained
 from hypothesis import given, strategies as st
 from random import Random
 import math