Simplify infinity check (#2587)

openmc-dev · Jul 5, 2023 · 21d8847 · 21d8847
1 parent 382bcb2
commit 21d8847
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openmc/universe.py b/openmc/universe.py
@@ -391,7 +391,7 @@ def plot(self, origin=None, width=None, pixels=40000,
 
         bb = self.bounding_box
         # checks to see if bounding box contains -inf or inf values
-        if np.isinf([bb[0][x], bb[1][x], bb[0][y], bb[1][y]]).any():
+        if np.isinf(bb.extent[basis]).any():
             if origin is None:
                 origin = (0, 0, 0)
             if width is None: