def update(self, model1, model2): """ Args: model1 - tuple containing (experiment, version, step) for the first compared model model2 - tuple containing (experiment, version, step) for the second compared model. """ success_map_1 = DataReader.get_episode_success_map(*model1) success_map_2 = DataReader.get_episode_success_map(*model2) result = success_map_1 + 2 * success_map_2 self.result_permutation = np.argsort(result) result = np.sort(result) self.episode_grid_heat_map.color = result.reshape(11, 51) self.color_result = result