type fix

TracePy-Org · Feb 3, 2025 · 2a953cd · 2a953cd
1 parent 6d897a9
commit 2a953cd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tracepy/ray.py b/tracepy/ray.py
@@ -39,9 +39,9 @@ def __init__(self, P: Union[List[List[float]], np.ndarray],
         N_rays = self.P.shape[0]
         # If wvl is a scalar, replicate for all rays.
         if np.isscalar(wvl):
-            self.wvl = np.full(N_rays, wvl)
+            self.wvl = np.full(N_rays, wvl, dtype=np.float64)
         else:
-            self.wvl = np.array(wvl)
+            self.wvl = np.asarray(wvl, dtype=np.float64)
         self.N = np.full(N_rays, N_0)
         self.active = np.ones(N_rays, dtype=bool)
         # Check that all direction vectors are normalized.