fix two moons simulated dtype

stefanradev93 · Jun 18, 2024 · 320f1ae · 320f1ae
1 parent 27f99cd
commit 320f1ae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_two_moons/conftest.py b/tests/test_two_moons/conftest.py
@@ -21,7 +21,7 @@ def contexts():
  return dict(r=r, alpha=alpha)
 
  def parameters():
- theta = np.random.uniform(-1.0, 1.0, size=2)
+ theta = np.random.uniform(-1.0, 1.0, size=2).astype(np.float32)
 
  return dict(theta=theta)