Skip to content

Commit 1c9500e

Browse files
committed
[ENH/CLN] Expose a few functions to the api
1 parent 2fb450f commit 1c9500e

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

gempy/API/__init__.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,8 @@
5959
remove_element_by_name
6060
)
6161

62+
from ..modules.serialization.save_load import save_model, load_model
63+
6264
# Geophysics
6365
from gempy_engine.modules.geophysics.gravity_gradient import calculate_gravity_gradient
6466

@@ -71,5 +73,6 @@
7173
'add_surface_points', 'add_orientations', 'delete_surface_points', 'delete_orientations',
7274
'create_orientations_from_surface_points_coords', 'modify_surface_points', 'modify_orientations',
7375
'add_structural_group', 'remove_structural_group_by_index', 'remove_structural_group_by_name', 'remove_element_by_name',
74-
'calculate_gravity_gradient'
76+
'calculate_gravity_gradient',
77+
'save_model', 'load_model'
7578
]

gempy/core/data/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
from .structural_group import FaultsRelationSpecialCase
1111
from ..color_generator import ColorsGenerator
1212

13+
from gempy_engine.core.data.interpolation_input import InterpolationInput
1314
from gempy_engine.core.data.stack_relation_type import StackRelationType
1415
from gempy_engine.core.data.options import InterpolationOptions
1516
from gempy_engine.core.data.solutions import Solutions
@@ -26,6 +27,6 @@
2627
'Grid', 'Topography',
2728
'ImporterHelper', 'GemPyEngineConfig', 'FaultsRelationSpecialCase', 'ColorsGenerator',
2829
# From gempy engine
29-
'StackRelationType', 'InterpolationOptions', 'Solutions', 'RawArraysSolution', 'GlobalAnisotropy', 'Transform',
30+
'InterpolationInput','StackRelationType', 'InterpolationOptions', 'Solutions', 'RawArraysSolution', 'GlobalAnisotropy', 'Transform',
3031
'FaultsData', 'FiniteFaultData', 'AvailableBackends', 'GeophysicsInput'
3132
]

0 commit comments

Comments
 (0)