Fix types to support numpy 2

equinor · Jun 28, 2024 · 8c23618 · 8c23618
1 parent 640de45
commit 8c23618
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/iterative_ensemble_smoother/esmda_inversion.py b/src/iterative_ensemble_smoother/esmda_inversion.py
@@ -98,7 +98,7 @@ def normalize_alpha(alpha: npt.NDArray[np.double]) -> npt.NDArray[np.double]:
     Examples
     --------
     >>> alpha = np.arange(10) + 1
-    >>> np.sum(1/normalize_alpha(alpha))
+    >>> float(np.sum(1/normalize_alpha(alpha)))
     1.0
     """
     factor = np.sum(1 / alpha)

diff --git a/src/iterative_ensemble_smoother/sies.py b/src/iterative_ensemble_smoother/sies.py
@@ -240,7 +240,7 @@ def sies_iteration(
         # Line 9
         N = self.X.shape[1]  # Ensemble members
         ans: npt.NDArray[np.double] = self.X + self.X @ self.W / np.sqrt(N - 1)
-        return ans
+        return ans.astype(responses.dtype)
 
     def propose_W(
         self, responses: npt.NDArray[np.double], step_length: float = 0.5