Change dtype to np.float64 in _polynomial_features.

optimagic-dev · Nov 4, 2022 · 7ed4c76 · 7ed4c76
1 parent 3007783
commit 7ed4c76
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/estimagic/optimization/tranquilo/fit_models.py b/src/estimagic/optimization/tranquilo/fit_models.py
@@ -447,7 +447,7 @@ def _polynomial_features(x, has_squares):
     else:
         n_poly_terms = n_interactions(n_params)
 
-    poly_terms = np.empty((n_poly_terms, n_samples), float)
+    poly_terms = np.empty((n_poly_terms, n_samples), np.float64)
     xt = x.T
 
     idx = 0