Cast points coordinates in gradient-boosted eqs

fatiando · Jun 18, 2024 · cf5ecb6 · cf5ecb6
1 parent 2f8e5b3
commit cf5ecb6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/harmonica/_equivalent_sources/gradient_boosted.py b/harmonica/_equivalent_sources/gradient_boosted.py
@@ -217,7 +217,9 @@ def fit(self, coordinates, data, weights=None):
             weights = weights.ravel()
         # Build point sources
         if self.points is None:
-            self.points_ = self._build_points(coordinates)
+            self.points_ = tuple(
+                p.astype(self.dtype) for p in self._build_points(coordinates)
+            )
         else:
             self.points_ = tuple(
                 p.astype(self.dtype) for p in vdb.n_1d_arrays(self.points, 3)