validate flux_type

illinois-ceesd · May 15, 2024 · dcd5812 · dcd5812
1 parent ca2f14e
commit dcd5812
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/advection.py b/examples/advection.py
@@ -42,8 +42,9 @@ def weak_flux(self, vec):
                 flux = self.dg.array_ops.stack((vec[:, 0], swp[:, 1]), axis=1)
             else:
                 flux = self.dg.array_ops.stack((swp[:, 0], vec[:, 1]), axis=1)
+        else:
+            raise ValueError("Invalid flux type")
 
-        # pylint: disable=used-before-assignment
         flux = flux * self.c * self.dg.normals
 
         return flux