Added modules __init__ and minor changes in json module

Author: flohorovicic

gempy/modules/__init__.py

@@ -0,0 +1,17 @@
+"""
+GemPy modules package.
+"""
+
+from . import json_io
+from . import custom_implicit_functions
+from . import data_manipulation
+from . import grids
+from . import advance_pile
+
+__all__ = [
+    'json_io',
+    'custom_implicit_functions',
+    'data_manipulation',
+    'grids',
+    'advance_pile'
+]

gempy/modules/json_io/json_operations.py

@@ -8,6 +8,7 @@
 import numpy as np
 
 from .schema import SurfacePoint, Orientation, GemPyModelJson
+from gempy_engine.core.data.stack_relation_type import StackRelationType
 
 
 class JsonIO:
@@ -75,9 +76,9 @@ def load_model_from_json(file_path: str):
             interpolation_options=interpolation_options
         )
 
-        # Map series to surfaces
-        for series in data['series']:
-            map_stack_to_surfaces(model, {series['name']: series['surfaces']})
+        # Map series to surfaces with structural relations
+        mapping_object = {series['name']: series['surfaces'] for series in data['series']}
+        map_stack_to_surfaces(model, mapping_object, series_data=data['series'])
 
         return model