Merge pull request #376 from MiraGeoscience/GEOPY-2795

GEOPY-2795: Update dictionary to choose driver from run_command

Author: domfournier

simpeg_drivers/__init__.py

@@ -131,7 +131,7 @@ def assets_path() -> Path:
     ),
     "joint surveys": (
         "simpeg_drivers.joint.joint_surveys",
-        {"inversion": "JointSurveyDriver"},
+        {"inversion": "JointSurveysDriver"},
     ),
     "magnetic scalar": (
         "simpeg_drivers.potential_fields.magnetic_scalar",

simpeg_drivers/joint/joint_cross_gradient/__init__.py

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

simpeg_drivers/joint/joint_petrophysics/__init__.py

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

simpeg_drivers/joint/joint_surveys/__init__.py

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

tests/uijson_test.py

@@ -16,6 +16,7 @@
 import numpy as np
 import pytest
 from geoapps_utils.driver.data import BaseData
+from geoapps_utils.run import load_ui_json_as_dict
 from geoh5py import Workspace
 from geoh5py.ui_json import InputFile
 from geoh5py.ui_json.annotations import Deprecated
@@ -357,3 +358,11 @@ def test_legacy_uijson(tmp_path: Path, caplog):
                     assert "The Batch2D classes will be deprecated" in caplog.text
 
             assert driver.models
+
+
+def test_driver_from_uijson():
+    path = Path(__file__).resolve().parent / "legacy/v0.2.1"
+
+    for file in path.glob("*.ui.json"):
+        input_file = load_ui_json_as_dict(file)
+        assert from_input_file(input_file)