Bug fix

aditiiyer · Nov 18, 2024 · f3b3fbd · f3b3fbd
1 parent d2f683a
commit f3b3fbd
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/cerr/dataclasses/dose.py b/cerr/dataclasses/dose.py
@@ -694,11 +694,12 @@ def sum(doseIndV, planC, fxCorrectDict={}):
     frxCorrectFlag = False
     fnHandle = None
     if len(fxCorrectDict) > 0:
+        dictCpy = fxCorrectDict.copy()
         frxCorrectFlag = True
         # Identify fractionation correction method
         methodDict = {"fractionNum": fractionNumCorrect,
                       "fractionSize": fractionSizeCorrect}
-        correctionType = fxCorrectDict.pop('correctionType')
+        correctionType = dictCpy.pop('correctionType')
         fnHandle = methodDict[correctionType]
 
     # Create shared grid from max extents of all dose grids
@@ -733,17 +734,17 @@ def sum(doseIndV, planC, fxCorrectDict={}):
         # Get dose array and grid extents
         doseArray = planC.dose[doseNum].doseArray
         doseGrid = origGridList[doseNum]
-        gridMatchFlag = ((doseGrid[0] == xOutV).all() and
-                         (doseGrid[1] == yOutV).all() and
-                         (doseGrid[2] == zOutV).all())
+        gridMatchFlag = (np.array_equal(doseGrid[0], xOutV) and
+                         np.array_equal(doseGrid[1], yOutV) and
+                         np.array_equal(doseGrid[2], zOutV))
 
         # Fraction correct
         if frxCorrectFlag:
             # Get fraction size
             frxSize = getFrxSize(doseNum, planC)
             # Fractionation correction
-            fxCorrectDict['inputFrxSize'] = frxSize
-            doseArray = fnHandle(doseArray, **fxCorrectDict)
+            dictCpy['inputFrxSize'] = frxSize
+            doseArray = fnHandle(doseArray, **dictCpy)
 
         # Resample to shared grid
         if gridMatchFlag: