Prevent ValueError in task initialisation

trustimaging · Nov 15, 2024 · 0f23542 · 0f23542
1 parent 3bb49b1
commit 0f23542
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/mosaic/core/task.py b/mosaic/core/task.py
@@ -693,7 +693,11 @@ def __init__(self, proxy, method, *args, **kwargs):
         self._remote = None
 
         self.state_changed('pending')
-        self.remote_runtime
+
+        try:
+            self.remote_runtime
+        except ValueError:
+            pass
 
     async def init(self):
         """

diff --git a/stride_examples/examples/homogeneous_elastic/forward_2D.py b/stride_examples/examples/homogeneous_elastic/forward_2D.py
@@ -96,7 +96,7 @@ async def main(runtime):
         sub_problem = problem.sub_problem(shot.id)
         shot_wavelets = sub_problem.shot.wavelets
 
-        pde.clear_operators()
+        await pde.clear_operators()
         traces = await pde(shot_wavelets, vp, vs, rho, problem=sub_problem, **config)
 
         data_stride = traces.data.copy()