bug fix

Author: KriFos1

ensemble/ensemble.py

@@ -527,11 +527,11 @@ def calc_ml_prediction(self, input_state=None):
                         batch_size = 1000
                     en_pred = []
                     batch_en = [np.arange(start, start + batch_size) for start in
-                                np.arange(0, self.ne - batch_size, batch_size)]
+                                np.arange(0, ml_ne - batch_size, batch_size)]
                     if len(batch_en): # if self.ne is less than batch_size
-                        batch_en.append(np.arange(batch_en[-1][-1]+1, self.ne))
+                        batch_en.append(np.arange(batch_en[-1][-1]+1, ml_ne))
                     else:
-                        batch_en.append(np.arange(0, self.ne))
+                        batch_en.append(np.arange(0, ml_ne))
                     for n_e in batch_en:
                         _ = [self.sim.run_fwd_sim(state, member_index, nosim=True) for state, member_index in
                                 zip([list_state[curr_n] for curr_n in n_e], [list_member_index[curr_n] for curr_n in n_e])]