[ENH] Making sure that compute_model_at only computes the custom grid

Author: Leguark

gempy/API/compute_API.py

@@ -82,11 +82,14 @@ def compute_model_at(gempy_model: GeoModel, at: np.ndarray,
     Returns:
         np.ndarray: The computed geological model at the specified coordinates.
     """
+    
+    print("WARNING: This function sets a custom grid and computes the model so be wary of side effects.")
     set_custom_grid(
         grid=gempy_model.grid,
-        xyz_coord=at
+        xyz_coord=at,
+        reset=True
     )
-
+    
     sol = compute_model(gempy_model, engine_config, validate_serialization=True)
     return sol.raw_arrays.custom
 

gempy/API/grid_API.py

@@ -88,11 +88,11 @@ def set_topography_from_file(grid: Grid, filepath: str, crop_to_extent: Union[Se
     return set_topography_from_subsurface_structured_grid(grid, struct)
 
 
-def set_custom_grid(grid: Grid, xyz_coord: np.ndarray):
+def set_custom_grid(grid: Grid, xyz_coord: np.ndarray, reset: bool = False):
     custom_grid = CustomGrid(values=xyz_coord)
     grid.custom_grid = custom_grid
 
-    set_active_grid(grid, [Grid.GridTypes.CUSTOM])
+    set_active_grid(grid, grid_type=[Grid.GridTypes.CUSTOM], reset=reset)
     return grid.custom_grid