[BUG-ish] Getting gempy_engine tests running

Author: Leguark

gempy_engine/API/interp_single/_octree_generation.py

@@ -18,20 +18,21 @@
 
 def interpolate_on_octree(interpolation_input: InterpolationInput, options: InterpolationOptions,
                           data_shape: InputDataDescriptor) -> OctreeLevel:
+    
     if BackendTensor.engine_backend is not AvailableBackends.PYTORCH and COMPUTE_GRADIENTS is False:
-        interpolation_input = copy.deepcopy(interpolation_input)
+        temp_interpolation_input = copy.deepcopy(interpolation_input)
 
     # * Interpolate - centers
-    output_0_centers: List[InterpOutput] = interpolate_all_fields(interpolation_input, options, data_shape)  # interpolate - centers
+    output_0_centers: List[InterpOutput] = interpolate_all_fields(temp_interpolation_input, options, data_shape)  # interpolate - centers
 
     # * Interpolate - corners
-    grid_0_centers: Grid = interpolation_input.grid  # ? This could be moved to the next section
+    grid_0_centers: Grid = temp_interpolation_input.grid  # ? This could be moved to the next section
     if options.compute_corners:
         grid_0_corners = Grid.from_xyz_coords(
             xyz_coords=_generate_corners(regular_grid=grid_0_centers.regular_grid)
         )
-        interpolation_input.grid = grid_0_corners  # * Prepare grid for next interpolation
-        output_0_corners: List[InterpOutput] = interpolate_all_fields(interpolation_input, options, data_shape)  # * This is unnecessary for the last level except for Dual contouring
+        temp_interpolation_input.grid = grid_0_corners  # * Prepare grid for next interpolation
+        output_0_corners: List[InterpOutput] = interpolate_all_fields(temp_interpolation_input, options, data_shape)  # * This is unnecessary for the last level except for Dual contouring
     else:
         output_0_corners = []
         grid_0_corners = None

gempy_engine/API/model/model_api.py

@@ -25,8 +25,7 @@ def compute_model(interpolation_input: InterpolationInput, options: Interpolatio
 
     # TODO: Make sure if this works with TF
     # ! If we inline this it seems the deepcopy does not work
-
-    if BackendTensor.engine_backend is AvailableBackends.PYTORCH and COMPUTE_GRADIENTS is False:
+    if BackendTensor.engine_backend is not AvailableBackends.PYTORCH and COMPUTE_GRADIENTS is False:
         interpolation_input = copy.deepcopy(interpolation_input)
 
     output: list[OctreeLevel] = interpolate_n_octree_levels(

tests/test_common/test_modules/test_geophysics.py

@@ -51,7 +51,7 @@ def test_gravity(simple_model_interpolation_input, n_oct_levels=3):
 
     print(solutions.gravity)
 
-    if plot_pyvista or True:
+    if plot_pyvista or False:
         pv.global_theme.show_edges = True
         p = pv.Plotter()
         plot_octree_pyvista(p, solutions.octrees_output, n_oct_levels -1)