Update flowpm/raytracing.py

dtype changed Co-authored-by: Francois Lanusse <[email protected]>
DifferentiableUniverseInitiative · Apr 28, 2021 · f65b113 · f65b113
1 parent b2e2d0e
commit f65b113
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flowpm/raytracing.py b/flowpm/raytracing.py
@@ -42,7 +42,7 @@ def random_2d_shift():
   I = np.zeros(3)
   facx = np.random.uniform(0, 1)
   facy = np.random.uniform(0, 1)
-  xyshifts = [np.asarray([facx, facy, 0], dtype=float)]
+  xyshifts = [np.asarray([facx, facy, 0], dtype=np.float32)]
   return I + xyshifts