GEOPY-2826: Plate match parallel process fails without good error log

simpeg_drivers/plate_simulation/__init__.py

@@ -7,5 +7,3 @@
 #  (see LICENSE file at the root of this source code package).                      '
 #                                                                                   '
 # '''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
-
-from .driver import PlateSimulationDriver

simpeg_drivers/plate_simulation/match/__init__.py

@@ -7,5 +7,3 @@
 #  (see LICENSE file at the root of this source code package).                      '
 #                                                                                   '
 # '''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
-
-from .driver import PlateMatchDriver

simpeg_drivers/plate_simulation/match/driver.py

@@ -15,6 +15,10 @@
 from pathlib import Path
 from typing import Self
 
+import matplotlib
+
+
+matplotlib.use("Agg")
 import matplotlib.pyplot as plt
 import numpy as np
 from dask.distributed import Client, Future, progress
@@ -194,6 +198,7 @@ def _create_plate_from_parameters(
         :return: MaxwellPlate object created from the parameters.
         """
         center = self.params.survey.vertices[index_center]
+
         center[2] = (
             self._drape_heights[index_center]
             - model_options.overburden_options.thickness
@@ -364,6 +369,10 @@ def run(self):
                 query, strike_angle[ii]
             )
             flip = is_up_dip(observed[:, indices])
+
+            if flip:
+                indices = np.flip(indices, axis=0)
+
             # Loop through files and compute scores and find the best match
             scores, centers = self.run_scores(spatial_projection, observed[:, indices])
             ranked = np.argsort(scores)
@@ -433,6 +442,9 @@ def run_scores(self, spatial_projection, data) -> tuple[np.ndarray, np.ndarray]:
         )
         tasks = []
         for file_batch in file_split:
+            if len(file_batch) == 0:
+                continue
+
             args = (
                 file_batch,
                 spatial_projection,

simpeg_drivers/plate_simulation/sweep/__init__.py

@@ -7,5 +7,3 @@
 #  (see LICENSE file at the root of this source code package).                      '
 #                                                                                   '
 # '''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''
-
-from .driver import PlateSweepDriver

tests/run_tests/driver_joint_pgi_homogeneous_test.py

@@ -223,7 +223,7 @@ def test_homogeneous_run(
 
                 # Remove inversion type as per current json on file
                 options = driver.out_group.options
-                del options["inversion_type"]
+                options.pop("inversion_type", None)
                 driver.out_group.options = options
                 drivers.append(driver)
             else: