Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion simpeg_drivers/components/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
from geoh5py.workspace import Workspace

from simpeg_drivers.components.meshes import InversionMesh
from simpeg_drivers.params import InversionBaseOptions
from simpeg_drivers.options import InversionBaseOptions

from copy import deepcopy
from re import findall
Expand Down
2 changes: 1 addition & 1 deletion simpeg_drivers/components/factories/misfit_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@

from simpeg_drivers.components.data import InversionData
from simpeg_drivers.components.meshes import InversionMesh
from simpeg_drivers.params import BaseOptions
from simpeg_drivers.options import BaseOptions

import numpy as np
from geoh5py.objects import Octree
Expand Down
2 changes: 1 addition & 1 deletion simpeg_drivers/components/factories/receiver_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
if TYPE_CHECKING:
from geoapps_utils.driver.params import BaseParams

from simpeg_drivers.params import BaseOptions
from simpeg_drivers.options import BaseOptions

import numpy as np
from geoapps_utils.utils.transformations import rotate_xyz
Expand Down
2 changes: 1 addition & 1 deletion simpeg_drivers/components/factories/simpeg_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
if TYPE_CHECKING:
from geoapps_utils.driver.params import BaseParams

from simpeg_drivers.params import BaseOptions
from simpeg_drivers.options import BaseOptions

# TODO Redesign simpeg factory to avoid pylint arguments-differ complaint

Expand Down
2 changes: 1 addition & 1 deletion simpeg_drivers/components/factories/simulation_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
if TYPE_CHECKING:
from geoapps_utils.driver.params import BaseParams

from simpeg_drivers.params import BaseOptions
from simpeg_drivers.options import BaseOptions

from pathlib import Path

Expand Down
2 changes: 1 addition & 1 deletion simpeg_drivers/components/factories/source_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@
if TYPE_CHECKING:
from geoapps_utils.driver.params import BaseParams

from simpeg_drivers.params import BaseOptions
from simpeg_drivers.options import BaseOptions


class SourcesFactory(SimPEGFactory):
Expand Down
2 changes: 1 addition & 1 deletion simpeg_drivers/components/factories/survey_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
if TYPE_CHECKING:
from geoapps_utils.driver.params import BaseParams

from simpeg_drivers.params import BaseOptions
from simpeg_drivers.options import BaseOptions

import numpy as np
import simpeg.electromagnetics.time_domain as tdem
Expand Down
2 changes: 1 addition & 1 deletion simpeg_drivers/components/locations.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
if TYPE_CHECKING:
from geoh5py.workspace import Workspace

from simpeg_drivers.params import BaseForwardOptions, BaseInversionOptions
from simpeg_drivers.options import BaseForwardOptions, BaseInversionOptions

import numpy as np
from geoh5py.objects import ObjectBase, Points
Expand Down
2 changes: 1 addition & 1 deletion simpeg_drivers/components/meshes.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
from octree_creation_app.utils import octree_2_treemesh, treemesh_2_octree
from scipy.sparse import csr_matrix, identity

from simpeg_drivers.params import BaseForwardOptions, BaseInversionOptions
from simpeg_drivers.options import BaseForwardOptions, BaseInversionOptions
from simpeg_drivers.utils.meshes import auto_mesh_parameters
from simpeg_drivers.utils.utils import drape_2_tensor

Expand Down
2 changes: 1 addition & 1 deletion simpeg_drivers/components/topography.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
from geoh5py.workspace import Workspace

from simpeg_drivers.components.meshes import InversionMesh
from simpeg_drivers.params import BaseOptions
from simpeg_drivers.options import BaseOptions

import warnings

Expand Down
2 changes: 1 addition & 1 deletion simpeg_drivers/components/windows.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
from geoapps_utils.driver.params import BaseParams
from geoh5py.workspace import Workspace

from simpeg_drivers.params import BaseForwardOptions, BaseInversionOptions
from simpeg_drivers.options import BaseForwardOptions, BaseInversionOptions

import numpy as np
from geoh5py.objects import Grid2D, PotentialElectrode
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -28,8 +28,8 @@
from geoapps_utils.driver.driver import BaseDriver
from geoh5py.data import FloatData

from simpeg_drivers.depth_of_investigation.sensitivity_cutoff.params import (
SensitivityCutoffParams,
from simpeg_drivers.depth_of_investigation.sensitivity_cutoff.options import (
SensitivityCutoffOptions,
)


Expand Down Expand Up @@ -107,9 +107,9 @@ class SensitivityCutoffDriver(BaseDriver):
below the provided cutoff percentage.
"""

_params_class = SensitivityCutoffParams
_params_class = SensitivityCutoffOptions

def __init__(self, params: SensitivityCutoffParams):
def __init__(self, params: SensitivityCutoffOptions):
super().__init__(params)

def run(self):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@
from simpeg_drivers import assets_path


class SensitivityCutoffParams(BaseData):
class SensitivityCutoffOptions(BaseData):
"""
Sensitivity cutoff parameters for depth of investigation studies.

Expand Down
8 changes: 4 additions & 4 deletions simpeg_drivers/driver.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,11 +59,11 @@
InversionWindow,
)
from simpeg_drivers.components.factories import DirectivesFactory, MisfitFactory
from simpeg_drivers.params import (
from simpeg_drivers.options import (
BaseForwardOptions,
BaseInversionOptions,
)
from simpeg_drivers.joint.params import BaseJointOptions
from simpeg_drivers.joint.options import BaseJointOptions
from simpeg_drivers.utils.utils import tile_locations
from simpeg_drivers.utils.regularization import cell_neighbors, set_rotated_operators

Expand All @@ -72,7 +72,7 @@


class InversionDriver(BaseDriver):
_params_class = BaseForwardOptions | BaseInversionOptions
_options_class = BaseForwardOptions | BaseInversionOptions
_inversion_type: str | None = None
_validations = None

Expand Down Expand Up @@ -582,7 +582,7 @@ def start(cls, filepath: str | Path, driver_class=None):
)

with ifile.data["geoh5"].open(mode="r+"):
params = driver_class._params_class.build(ifile)
params = driver_class._options_class.build(ifile)
driver = driver_class(params)

driver.run()
Expand Down
2 changes: 1 addition & 1 deletion simpeg_drivers/electricals/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
DC3DForwardOptions,
DC3DInversionOptions,
)
from .induced_polarization.three_dimensions.params import (
from .induced_polarization.three_dimensions.options import (
IP3DForwardOptions,
IP3DInversionOptions,
)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,11 +11,11 @@

from __future__ import annotations

from simpeg_drivers.electricals.direct_current.pseudo_three_dimensions.params import (
from simpeg_drivers.electricals.direct_current.pseudo_three_dimensions.options import (
DCBatch2DForwardOptions,
DCBatch2DInversionOptions,
)
from simpeg_drivers.electricals.direct_current.two_dimensions.params import (
from simpeg_drivers.electricals.direct_current.two_dimensions.options import (
DC2DForwardOptions,
DC2DInversionOptions,
)
Expand All @@ -25,14 +25,14 @@
class DCBatch2DForwardDriver(BaseBatch2DDriver):
"""Direct Current batch 2D forward driver."""

_params_class = DCBatch2DForwardOptions
_options_class = DCBatch2DForwardOptions
_params_2d_class = DC2DForwardOptions
_validations = None


class DCBatch2DInversionDriver(BaseBatch2DDriver):
"""Direct Current batch 2D inversion driver."""

_params_class = DCBatch2DInversionOptions
_options_class = DCBatch2DInversionOptions
_params_2d_class = DC2DInversionOptions
_validations = None
Original file line number Diff line number Diff line change
Expand Up @@ -18,10 +18,10 @@
from geoh5py.objects import Octree, PotentialElectrode

from simpeg_drivers import assets_path
from simpeg_drivers.electricals.params import (
from simpeg_drivers.electricals.options import (
FileControlOptions,
)
from simpeg_drivers.params import (
from simpeg_drivers.options import (
BaseForwardOptions,
BaseInversionOptions,
DrapeModelOptions,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
# '''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''


from .params import DC3DForwardOptions, DC3DInversionOptions
from .options import DC3DForwardOptions, DC3DInversionOptions

# pylint: disable=unused-import
# flake8: noqa
Original file line number Diff line number Diff line change
Expand Up @@ -13,18 +13,18 @@

from simpeg_drivers.driver import InversionDriver

from .params import DC3DForwardOptions, DC3DInversionOptions
from .options import DC3DForwardOptions, DC3DInversionOptions


class DC3DForwardDriver(InversionDriver):
"""Direct Current 3D forward driver."""

_params_class = DC3DForwardOptions
_options_class = DC3DForwardOptions
_validation = None


class DC3DInversionDriver(InversionDriver):
"""Direct Current 3D inversion driver."""

_params_class = DC3DInversionOptions
_options_class = DC3DInversionOptions
_validation = None
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
from geoh5py.data import FloatData

from simpeg_drivers import assets_path
from simpeg_drivers.params import BaseForwardOptions, BaseInversionOptions
from simpeg_drivers.options import BaseForwardOptions, BaseInversionOptions


class DC3DForwardOptions(BaseForwardOptions):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
# '''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''


from .params import DC2DForwardOptions, DC2DInversionOptions
from .options import DC2DForwardOptions, DC2DInversionOptions


__all__ = ["DC2DForwardOptions", "DC2DInversionOptions"]
Original file line number Diff line number Diff line change
Expand Up @@ -13,18 +13,18 @@

from simpeg_drivers.electricals.driver import Base2DDriver

from .params import DC2DForwardOptions, DC2DInversionOptions
from .options import DC2DForwardOptions, DC2DInversionOptions


class DC2DForwardDriver(Base2DDriver):
"""Direct Current 2D forward driver."""

_params_class = DC2DForwardOptions
_options_class = DC2DForwardOptions
_validations = None


class DC2DInversionDriver(Base2DDriver):
"""Direct Current 2D inversion driver."""

_params_class = DC2DInversionOptions
_options_class = DC2DInversionOptions
_validations = None
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
from pydantic import BaseModel, ConfigDict, field_validator, model_validator

from simpeg_drivers import assets_path
from simpeg_drivers.params import (
from simpeg_drivers.options import (
BaseForwardOptions,
BaseInversionOptions,
DrapeModelOptions,
Expand Down
4 changes: 2 additions & 2 deletions simpeg_drivers/electricals/driver.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@
from simpeg_drivers.components.windows import InversionWindow
from simpeg_drivers.driver import InversionDriver
from simpeg_drivers.line_sweep.driver import LineSweepDriver
from simpeg_drivers.params import (
from simpeg_drivers.options import (
BaseForwardOptions,
BaseInversionOptions,
LineSelectionOptions,
Expand Down Expand Up @@ -80,7 +80,7 @@ def create_drape_mesh(self) -> DrapeModel:
class BaseBatch2DDriver(LineSweepDriver):
"""Base class for batch 2D DC and IP forward and inversion drivers."""

_params_class: type[BaseForwardOptions | BaseInversionOptions]
_options_class: type[BaseForwardOptions | BaseInversionOptions]
_params_2d_class: type[BaseForwardOptions | BaseInversionOptions]
_validations = None
_model_list: list[str] = []
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,11 +12,11 @@
from __future__ import annotations

from simpeg_drivers.electricals.driver import BaseBatch2DDriver
from simpeg_drivers.electricals.induced_polarization.pseudo_three_dimensions.params import (
from simpeg_drivers.electricals.induced_polarization.pseudo_three_dimensions.options import (
IPBatch2DForwardOptions,
IPBatch2DInversionOptions,
)
from simpeg_drivers.electricals.induced_polarization.two_dimensions.params import (
from simpeg_drivers.electricals.induced_polarization.two_dimensions.options import (
IP2DForwardOptions,
IP2DInversionOptions,
)
Expand All @@ -25,7 +25,7 @@
class IPBatch2DForwardDriver(BaseBatch2DDriver):
"""Induced Polarization batch 2D forward driver."""

_params_class = IPBatch2DForwardOptions
_options_class = IPBatch2DForwardOptions
_params_2d_class = IP2DForwardOptions
_validations = None
_model_list = ["conductivity_model"]
Expand All @@ -34,7 +34,7 @@ class IPBatch2DForwardDriver(BaseBatch2DDriver):
class IPBatch2DInversionDriver(BaseBatch2DDriver):
"""Induced Polarization batch 2D inversion driver."""

_params_class = IPBatch2DInversionOptions
_options_class = IPBatch2DInversionOptions
_params_2d_class = IP2DInversionOptions
_validations = None
_model_list = ["conductivity_model"]
Original file line number Diff line number Diff line change
Expand Up @@ -18,10 +18,10 @@
from geoh5py.objects import Octree, PotentialElectrode

from simpeg_drivers import assets_path
from simpeg_drivers.electricals.params import (
from simpeg_drivers.electricals.options import (
FileControlOptions,
)
from simpeg_drivers.params import (
from simpeg_drivers.options import (
BaseForwardOptions,
BaseInversionOptions,
DrapeModelOptions,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
# '''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''''


from .params import (
from .options import (
IP3DForwardOptions,
IP3DInversionOptions,
)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@

from simpeg_drivers.driver import InversionDriver

from .params import (
from .options import (
IP3DForwardOptions,
IP3DInversionOptions,
)
Expand All @@ -22,12 +22,12 @@
class IP3DForwardDriver(InversionDriver):
"""Induced Polarization 3D forward driver."""

_params_class = IP3DForwardOptions
_options_class = IP3DForwardOptions
_validations = None


class IP3DInversionDriver(InversionDriver):
"""Induced Polarization 3D inversion driver."""

_params_class = IP3DInversionOptions
_options_class = IP3DInversionOptions
_validations = None
Loading
Loading