fix broken test due to class name change

wilsonmr · Apr 26, 2021 · f215adb · f215adb · wilsonmr · Apr 26, 2021
1 parent 1548be0
commit f215adb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/anvil/tests/test_distributions.py b/anvil/tests/test_distributions.py
@@ -5,7 +5,7 @@
 from math import sqrt
 
 import numpy as np
-from anvil.distributions import NormalDist
+from anvil.distributions import Gaussian
 
 MEAN = 0
 SIGMA = 1
@@ -20,7 +20,7 @@ def test_normal_distribution():
 
     """
     lattice_size = 5
-    generator = NormalDist(lattice_size, sigma=SIGMA, mean=MEAN)
+    generator = Gaussian(lattice_size, sigma=SIGMA, mean=MEAN)
     sample_pt, _ = generator(N_SAMPLE)
     sample_np = sample_pt.detach().numpy()
     np.testing.assert_allclose(