[ENH] Added section plots to sprember

Author: Leguark

gempy/API/grid_API.py

@@ -12,12 +12,11 @@
 def set_section_grid(grid: Grid, section_dict: dict):
     if grid.sections is None:
         grid.sections = Sections(regular_grid=grid.regular_grid, section_dict=section_dict)
-        grid.sections
     else:
         grid.sections.set_sections(section_dict,
                                    regular_grid=grid.regular_grid)
 
-    set_active_grid(grid, [GridTypes.SECTIONS])
+    set_active_grid(grid, [Grid.GridTypes.SECTIONS])
     return grid.sections
 
 

gempy/core/data/geo_model.py

@@ -233,7 +233,7 @@ def project_bounds(self) -> np.ndarray:
         return self.grid.bounding_box
 
     @property
-    def extent_transformed(self) -> np.ndarray:
+    def extent_transformed_transformed_by_input(self) -> np.ndarray:
         transformed = self.input_transform.apply(self.project_bounds)  # ! grid already has the grid transform applied
         new_extents = np.array([transformed[:, 0].min(), transformed[:, 0].max(),
                                 transformed[:, 1].min(), transformed[:, 1].max(),

gempy/modules/data_manipulation/engine_factory.py

@@ -11,9 +11,12 @@
 from gempy_engine.core.data.transforms import Transform
 
 
-def interpolation_input_from_structural_frame(geo_model: "GeoModel") -> InterpolationInput:
+def interpolation_input_from_structural_frame(geo_model: "gempy.data.GeoModel") -> InterpolationInput:
+    import gempy # ! This is important for type safety
+    geo_model: gempy.data.GeoModel = geo_model
+
     _legacy_factor = 0
-    
+
     structural_frame: StructuralFrame = geo_model.structural_frame
     input_transform: Transform = geo_model.input_transform
     grid: Grid = geo_model.grid
@@ -36,7 +39,7 @@ def interpolation_input_from_structural_frame(geo_model: "GeoModel") -> Interpol
     grid: engine_grid.EngineGrid = _apply_input_transform_to_grids(
         grid=grid,
         input_transform=input_transform,
-        extent_transformed=geo_model.extent_transformed
+        extent_transformed=geo_model.extent_transformed_transformed_by_input
     )
 
     interpolation_input: InterpolationInput = InterpolationInput(