add try except for getting riemann solver

python-hydro · Aug 1, 2024 · 2bc645b · 2bc645b
1 parent 172f62c
commit 2bc645b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pyro/compressible/simulation.py b/pyro/compressible/simulation.py
@@ -112,7 +112,11 @@ def initialize(self, extra_vars=None, ng=4):
         my_data = self.data_class(my_grid)
 
         # Make sure we use CGF for riemann solver when we do SphericalPolar
-        riemann = self.rp.get_param("compressible.riemann")
+        try:
+            riemann = self.rp.get_param("compressible.riemann")
+        except KeyError:
+            msg.fail("ERROR: Riemann Solver is not set.")
+
         if my_grid.coord_type == 1 and riemann == "HLLC":
             msg.fail("ERROR: Only CGF Riemann Solver is supported " +
                      "with SphericalPolar Geometry")