Design points can be passed directly as an array or DataFrame

corbass/inversion.py

@@ -355,7 +355,18 @@ class Inversion:
             The posterior covariance of the field at design points.
         """
         # Setup design points
-        self.x_desi, self.n_desi = self.desi_points(n_desi)
+        if isinstance(n_desi, np.ndarray):
+            self.x_desi = n_desi
+            self.n_desi = self.x_desi.shape[1]
+        elif isinstance(n_desi, DataFrame):
+            self.n_desi = len(n_desi)
+            self.x_desi = np.zeros(4, self.n_desi)
+            self.x_desi[0] = n_desi[['co-lat']]
+            self.x_desi[1] = n_desi[['lon']]
+            self.x_desi[2] = n_desi[['rad']]
+            self.x_desi[3] = n_desi[['t']]
+        else:
+            self.x_desi, self.n_desi = self.desi_points(n_desi)
 
         # Allocate and retrieve basis functions at design points
         dspharm_desi = np.empty((3, 3*self.n_desi), order='F')