set restart level to root in batchwise scheme after material critical…

…ity search
openmsr · Oct 30, 2023 · a9e9133 · a9e9133
1 parent 2591bb9
commit a9e9133
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/openmc/deplete/batchwise.py b/openmc/deplete/batchwise.py
@@ -1547,7 +1547,8 @@ def search_for_keff(self, x, step_index):
 
             # restart level and perform dilution
             self.bw_geom._set_cell_attrib(self.restart_level)
-            x, root = self.bw_mat.search_for_keff(x, step_index)
+            x, _ = self.bw_mat.search_for_keff(x, step_index)
+            root = self.restart_level
             #update dulution interval
             #if step_index == self.dilute_interval:
             #    self.dilute_interval += self.step_interval
@@ -1646,10 +1647,12 @@ def search_for_keff(self, x, step_index):
         """
         #Start by doing a geometrical search§
         x, root = self.bw_geom.search_for_keff(x, step_index)
+
         #check if upper geometrical limit gets hit
         if root >= self.bw_geom.bracket_limit[1]:
             # Reset geometry and refuel
             self.bw_geom._set_cell_attrib(self.restart_level)
-            x,root = self.bw_mat.search_for_keff(x, step_index)
-
-        return x,root
+            x,_ = self.bw_mat.search_for_keff(x, step_index)
+            root = self.restart_level
+
+        return x, root