add verbose assert

AlexHls · Oct 31, 2023 · 3ac97e0 · 3ac97e0
1 parent cfce38d
commit 3ac97e0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tardis/plasma/properties/nlte_rate_equation_solver.py b/tardis/plasma/properties/nlte_rate_equation_solver.py
@@ -125,7 +125,9 @@ def calculate(
                 ),
                 jac=True,
             )
-            assert solution.success
+            assert solution.success, (
+                "No solution for NLTE population equation found or solver takes too long to converge"
+            )
             ion_number_density_nlte[shell] = solution.x[:-1]
             electron_densities_nlte[shell] = solution.x[-1]
         # TODO: change the jacobian and rate matrix to use shell id and get coefficients from the attribute of the class.