Allow csr matric for transform

Author: domfournier

simpeg/directives/_save_geoh5.py

@@ -4,7 +4,7 @@
 from pathlib import Path
 
 import numpy as np
-
+from scipy.sparse import csc_matrix, csr_matrix
 from .directives import InversionDirective
 from simpeg.maps import IdentityMap
 
@@ -188,7 +188,12 @@ def transforms(self, funcs: list | tuple):
 
         for fun in funcs:
             if not any(
-                [isinstance(fun, (IdentityMap, np.ndarray, float)), callable(fun)]
+                [
+                    isinstance(
+                        fun, (IdentityMap, np.ndarray, csr_matrix, csc_matrix, float)
+                    ),
+                    callable(fun),
+                ]
             ):
                 raise TypeError(
                     "Input transformation must be of type"
@@ -212,7 +217,9 @@ def apply_transformations(self, prop: np.ndarray) -> np.ndarray:
         """
         prop = prop.flatten()
         for fun in self.transforms:
-            if isinstance(fun, (IdentityMap, np.ndarray, float)):
+            if isinstance(
+                fun, (IdentityMap, np.ndarray, csr_matrix, csc_matrix, float)
+            ):
                 prop = fun * prop
             else:
                 prop = fun(prop)