From e4c332360175b66c8261c9d41643913cb9958e65 Mon Sep 17 00:00:00 2001
From: Gavin Ridley <gavin.keith.ridley@gmail.com>
Date: Mon, 17 Apr 2023 12:15:29 -0400
Subject: [PATCH] remove unnecessary error checking

Co-authored-by: Jonathan Shimwell <drshimwell@gmail.com>
---
 openmc/universe.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/openmc/universe.py b/openmc/universe.py
index 32768c305f..a4efb5f732 100644
--- a/openmc/universe.py
+++ b/openmc/universe.py
@@ -409,10 +409,8 @@ class Universe(UniverseBase):
 
                 if color_by == "cell":
                     expected_key_type = openmc.Cell
-                elif color_by == "material":
-                    expected_key_type = openmc.Material
                 else:
-                    raise Exception("wtf?")
+                    expected_key_type = openmc.Material
 
                 patches = []
                 for key, color in plot.colors.items():