pygeoinf 1.3.3__tar.gz → 1.3.4__tar.gz

{pygeoinf-1.3.3 → pygeoinf-1.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygeoinf
-Version: 1.3.3
+Version: 1.3.4
 Summary: A package for solving geophysical inference and inverse problems
 License: BSD-3-Clause
 License-File: LICENSE

{pygeoinf-1.3.3 → pygeoinf-1.3.4}/pygeoinf/checks/hilbert_space.py

@@ -198,4 +198,4 @@ class HilbertSpaceAxiomChecks:
             self._check_inplace_operations(x, y, a)
             self._check_copy(x)
 
-        print(f"✅ All {n_checks} Hilbert space axiom checks passed successfully.")
+        print(f"[✓] All {n_checks} Hilbert space axiom checks passed successfully.")

{pygeoinf-1.3.3 → pygeoinf-1.3.4}/pygeoinf/checks/linear_operators.py

@@ -194,4 +194,4 @@ class LinearOperatorAxiomChecks(NonLinearOperatorAxiomChecks):
                     self, op2, x1, y, a, check_rtol=check_rtol, check_atol=check_atol
                 )
 
-        print(f"✅ All {n_checks} linear operator checks passed successfully.")
+        print(f"[✓] All {n_checks} linear operator checks passed successfully.")

{pygeoinf-1.3.3 → pygeoinf-1.3.4}/pygeoinf/checks/nonlinear_operators.py

@@ -195,4 +195,4 @@ class NonLinearOperatorAxiomChecks:
                     self, op2, x, v, check_rtol=check_rtol, check_atol=check_atol
                 )
 
-        print(f"✅ All {n_checks} non-linear operator checks passed successfully.")
+        print(f"[✓] All {n_checks} non-linear operator checks passed successfully.")

{pygeoinf-1.3.3 → pygeoinf-1.3.4}/pygeoinf/hilbert_space.py

@@ -28,6 +28,7 @@ from abc import ABC, abstractmethod
 from typing import (
     TypeVar,
     List,
+    Union,
     Optional,
     Any,
     TYPE_CHECKING,
@@ -607,6 +608,50 @@ class EuclideanSpace(HilbertSpace):
         """
         return isinstance(x, np.ndarray) and x.shape == (self.dim,)
 
+    def subspace_projection(self, indices: Union[int, List[int]]) -> "LinearOperator":
+        """
+        Returns a projection operator onto specified coordinates.
+
+        This creates a linear operator that extracts the components at the given
+        indices, projecting from this space to a lower-dimensional Euclidean space.
+
+        Args:
+            indices: Single index or list of indices to project onto (0-indexed).
+
+        Returns:
+            LinearOperator from this space to EuclideanSpace(len(indices)).
+
+        Raises:
+            IndexError: If any index is out of range for this space's dimension.
+        """
+        from .linear_operators import LinearOperator
+
+        if isinstance(indices, int):
+            indices = [indices]
+
+        indices_array = np.array(indices)
+        if np.any(indices_array < 0) or np.any(indices_array >= self.dim):
+            raise IndexError(
+                f"Indices {indices_array} out of range for dimension {self.dim}"
+            )
+
+        target_space = EuclideanSpace(len(indices))
+
+        def forward(x: np.ndarray) -> np.ndarray:
+            return x[indices_array]
+
+        def adjoint_mapping(y: np.ndarray) -> np.ndarray:
+            result = np.zeros(self.dim)
+            result[indices_array] = y
+            return result
+
+        return LinearOperator(
+            self,
+            target_space,
+            forward,
+            adjoint_mapping=adjoint_mapping,
+        )
+
 
 class MassWeightedHilbertSpace(HilbertSpace):
     """

{pygeoinf-1.3.3 → pygeoinf-1.3.4}/pygeoinf/plot.py

@@ -247,7 +247,7 @@ def plot_corner_distributions(
                 sigma = np.sqrt(cov_posterior[i, i])
                 
                 # Create x-axis range
-                x = np.linspace(mu - 4 * sigma, mu + 4 * sigma, 200)
+                x = np.linspace(mu - 3.75 * sigma, mu + 3.75 * sigma, 200)
                 pdf = stats.norm.pdf(x, mu, sigma)
                 
                 # Plot the PDF
@@ -276,10 +276,10 @@ def plot_corner_distributions(
                 sigma_j = np.sqrt(cov_posterior[j, j])
                 sigma_i = np.sqrt(cov_posterior[i, i])
                 
-                x_range = np.linspace(mean_2d[0] - 3.5 * sigma_j, 
-                                    mean_2d[0] + 3.5 * sigma_j, 100)
-                y_range = np.linspace(mean_2d[1] - 3.5 * sigma_i, 
-                                    mean_2d[1] + 3.5 * sigma_i, 100)
+                x_range = np.linspace(mean_2d[0] - 3.75 * sigma_j, 
+                                    mean_2d[0] + 3.75 * sigma_j, 100)
+                y_range = np.linspace(mean_2d[1] - 3.75 * sigma_i, 
+                                    mean_2d[1] + 3.75 * sigma_i, 100)
                 
                 X, Y = np.meshgrid(x_range, y_range)
                 pos = np.dstack((X, Y))

{pygeoinf-1.3.3 → pygeoinf-1.3.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pygeoinf"
-version = "1.3.3"
+version = "1.3.4"
 description = "A package for solving geophysical inference and inverse problems"
 authors = ["David Al-Attar and Dan Heathcote"]
 readme = "README.md"