Adding changes to take off transitions code to save memory

Author: kanishkan91

demeter/change/expansion.py

@@ -46,7 +46,8 @@ def extense_parallel_helper(regix_metix, log, c, allregnumber, allregmet, spat_l
     # transitions[reg_met_mask, :, :] += trans_mat
 
     # calculate non-achieved change
-    transitions[reg_met_mask, :, :] += trans_mat
+    if transitions.size != 1:
+        transitions[reg_met_mask, :, :] += trans_mat
 
     non_chg = np.sum(abs(target_change[:, :, :])) / 2.
 

demeter/change/intensification.py

@@ -63,7 +63,8 @@ def intense_parallel_helper(regix_metix, spat_region, order_rules, allregnumber,
     # arr_reshaped = trans_mat.reshape(trans_mat.shape[0], -1)
     # np.savetxt("test.csv", arr_reshaped, delimiter=",")
     # log transition
-    transitions[reg_met_mask, :, :] += trans_mat
+    if transitions.size != 1:
+        transitions[reg_met_mask, :, :] += trans_mat
 
     # calculate non-achieved change
 

demeter/process.py

@@ -64,7 +64,10 @@ def prep_step(self):
                                                            self.s.lat, self.s.lon, self.step, self.s.kernel_vector,
                                                            self.s.weights, self.s.spat_ludataharm)
 
-        self.transitions = np.zeros(shape=(self.l_spat_region, self.l_order_rules, self.l_order_rules))
+        if self.config.save_transitions == 1:
+            self.transitions = np.zeros(shape=(self.l_spat_region, self.l_order_rules, self.l_order_rules))
+        else:
+            self.transitions = np.zeros(shape=1)
 
     def intense_pass(self, pass_num):
         """Conduct the first pass of intensification."""