Don't require interal_data for stepper solver

aragilar · Dec 7, 2018 · bf4046f · bf4046f
1 parent 8a1671f
commit bf4046f
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/disc_solver/solve/stepper.py b/src/disc_solver/solve/stepper.py
@@ -184,9 +184,12 @@ def v_θ_deriv_splitter(soln, cutoff=DEFAULT_SPLITTER_CUTOFF, **kwargs):
     """
     # pylint: disable=unused-argument
     v_θ = soln.solution[:, ODEIndex.v_θ]
-    problems = soln.internal_data.problems
-    if any("negative velocity" in pl for pl in problems.values()):
-        return SplitterStatus.SIGN_FLIP
+    if soln.internal_data is not None:
+        problems = soln.internal_data.problems
+        if any("negative velocity" in pl for pl in problems.values()):
+            return SplitterStatus.SIGN_FLIP
+    else:
+        log.notice("Skipping checking problems due to no internal data")
 
     v_θ_near_sonic = np_and(v_θ > cutoff, v_θ < 1)
     if not np_any(v_θ_near_sonic):