ci: Grab correct source from data repo

Author: JDBetteridge

.github/workflows/examples.yml

@@ -36,6 +36,12 @@ jobs:
     - name: Checkout devito
       uses: actions/checkout@v5
 
+    - name: Checkout data
+      uses: actions/checkout@v5
+      with:
+       repository: 'devitocodes/data'
+       path: 'data'
+
     - name: Setup conda
       uses: conda-incubator/setup-miniconda@v3
       with:

examples/timestepping/acoustic_superstep.py

@@ -135,10 +135,14 @@ def acoustic_model(model, t0, t1, t2, critical_dt, source, step=1, snapshots=1):
         t2 = 0.5
         critical_dt = 0.0013728
         zlim = 20
+        try:
+            path = f'{os.environ["VIRTUAL_ENV"]}/src'
+        except KeyError:
+            path = str(os.environ['GITHUB_WORKSPACE'])
         tmp_model = demo_model(
             'marmousi-isotropic',
             space_order=2,
-            data_path=f'{os.environ["VIRTUAL_ENV"]}/src/data',
+            data_path=f'{path}/data',
             nbl=0
         )
         cropped = tmp_model.vp.data[400:701, -321:-20]