def update(self, experiment): result = DataReader.get_episodes_success_counts(experiment) self.result_permutation = np.argsort(result) result = np.sort(result) self.episode_grid_heat_map.color = result.reshape(11, 51) self.color_result = result