DO NOT SQUASH: Simple accurate vp output

.github/workflows/python-tests.yml

@@ -148,6 +148,7 @@ jobs:
           mpiexec -n 6 pytest tests/parallel/test_fwi.py
           mpiexec -n 3 pytest tests/parallel/test_supershot_grad.py
           mpiexec -n 2 pytest tests/parallel/test_gradient_serialshots.py
+          mpiexec -n 6 pytest tests/parallel/test_simple_fwi_with_segy_output.py
 
     - name: Covering parallel 3D forward test
       continue-on-error: true
@@ -180,7 +181,11 @@ jobs:
     - name: Covering spatially parallelized shots in serial
       continue-on-error: true
       run: |
-          mpiexec -n 2 pytest --cov-report=xml --cov-append --cov=spyro tests/parallel/test_test_gradient_serialshots.py
+          mpiexec -n 2 pytest --cov-report=xml --cov-append --cov=spyro tests/parallel/test_gradient_serialshots.py
+    - name: Covering source parallelized fwi with segy output
+      continue-on-error: true
+      run: |
+          mpiexec -n 2 pytest --cov-report=xml --cov-append --cov=spyro tests/parallel/test_simple_fwi_with_segy_output.py
     - name: Uploading coverage to Codecov
       run: |
         export CODECOV_TOKEN="6cd21147-54f7-4b77-94ad-4b138053401d" && curl -s https://codecov.io/bash | bash

spyro/io/__init__.py

@@ -1,28 +1,30 @@
 from .basicio import (
     write_function_to_grid,
-    is_owner,
     save_shots,
     load_shots,
     read_mesh,
     interpolate,
     project_grid_velocity_data,
     # ensemble_forward_ad,
     # ensemble_forward_elastic_waves,
-    ensemble_gradient,
     # ensemble_gradient_elastic_waves,
     parallel_print,
     saving_source_and_receiver_location_in_csv,
-    switch_serial_shot,
-    ensemble_save,
-    ensemble_load,
-    delete_tmp_files,
-    ensemble_shot_record,
-    ensemble_functional,
     read_segy_velocity_model,
     read_bin_velocity_model,
     write_velocity_model
 )
-from .segy_io import create_segy, create_segy_from_grid
+from .parallelism_wrappers import (
+    delete_tmp_files,
+    ensemble_functional,
+    ensemble_gradient,
+    ensemble_load,
+    ensemble_save,
+    ensemble_shot_record,
+    is_owner,
+    switch_serial_shot,
+)
+from .segy_io import create_segy, create_segy_from_grid, export_scalar_field
 from .model_parameters import Model_parameters
 from .backwards_compatibility_io import Dictionary_conversion
 from . import dictionaryio
@@ -63,4 +65,5 @@
     "read_segy_velocity_model",
     "read_bin_velocity_model",
     "write_velocity_model",
+    "export_scalar_field"
 ]