Method call_wrap

AgentBench.old/src/tasks/alfworld/task.py:118–123

(self, parameters)

        return results

    def call_wrap(self, parameters):
        data_item, agent, index = parameters
        session = agent.create_session()
        result = self.predict_single(session, data_item, agent.name)
        self.save_single(index, data_item, result, session)
        return result

    def predict_single(self, session: Session, data_item: str, agent_name: str):
        env = SingleAlfredTWEnv(self.config, data_item)