def get_results(self):
    output = [{'instr_id': k, 'trajectory': v} for (k, v) in self.results.items()]
    return output
def get_results(self):
    output = [{'instr_id': k, 'trajectory': v} for (k, v) in self.results.items()]
    return output