Merge pull request #416 from sjjamsa/415-phi-dimension-3-letters

[BUG] Try using the whole string if getting PlotGrid fails using only the first letter. (issue #415)
CORE-GATECH-GROUP · Aug 22, 2020 · ee88bf4 · ee88bf4
2 parents 32458a3 + 87d5f58
commit ee88bf4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/serpentTools/detectors.py b/serpentTools/detectors.py
@@ -463,7 +463,11 @@ def _getSlices(self, fixed):
         return slices
 
     def _getPlotGrid(self, qty):
+        # Try with the first letter only
         grids = self.grids.get(qty[0].upper())
+        if grids is None:
+            # The phi-grid has three letters!
+            grids = self.grids.get(qty.upper())
         if grids is not None:
             return hstack((grids[:, 0], grids[-1, 1]))
         nItems = self.tallies.shape[self.indexes.index(qty)]