From 61316c3a75c62f28fa18feccc05adcb8acfe411f Mon Sep 17 00:00:00 2001 From: Jimmy Wei Date: Mon, 25 Jul 2022 17:33:00 -0400 Subject: [PATCH 1/3] Update parlai_chat_agent_state.py --- .../blueprints/parlai_chat/parlai_chat_agent_state.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_agent_state.py b/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_agent_state.py index 9a0004b2e..c410dbdce 100644 --- a/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_agent_state.py +++ b/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_agent_state.py @@ -57,7 +57,12 @@ def _load_data(self) -> None: self.messages = state["outputs"]["messages"] self.init_data = state["inputs"] self.final_submission = state["outputs"].get("final_submission") - self.metadata = _AgentStateMetadata(**state["metadata"]) + if 'metadata' in state: + self.metadata = _AgentStateMetadata(**state["metadata"]) + elif 'times' in state: + self.metadata = _AgentStateMetadata(start_time=state['times']['start_time'], end_time=state['times']['end_time']) + else: + self.meatadata = _AgentStateMetadata() def get_data(self) -> Dict[str, Any]: """Return dict with the messages of this agent""" From d3717de825c56d9f7db6575c89b2ed4298805d7e Mon Sep 17 00:00:00 2001 From: Jack Urbanek Date: Tue, 26 Jul 2022 12:33:33 -0400 Subject: [PATCH 2/3] black --- .../blueprints/parlai_chat/parlai_chat_agent_state.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_agent_state.py b/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_agent_state.py index c410dbdce..bcf867b43 100644 --- a/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_agent_state.py +++ b/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_agent_state.py @@ -60,7 +60,10 @@ def _load_data(self) -> None: if 'metadata' in state: self.metadata = _AgentStateMetadata(**state["metadata"]) elif 'times' in state: - self.metadata = _AgentStateMetadata(start_time=state['times']['start_time'], end_time=state['times']['end_time']) + self.metadata = _AgentStateMetadata( + start_time=state['times']['start_time'], + end_time=state['times']['end_time'], + ) else: self.meatadata = _AgentStateMetadata() From 57ffefde10f0f1de62b62633538c2bbcca54face Mon Sep 17 00:00:00 2001 From: Jack Urbanek Date: Tue, 26 Jul 2022 12:36:14 -0400 Subject: [PATCH 3/3] "meatadata" fix --- .../blueprints/parlai_chat/parlai_chat_agent_state.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_agent_state.py b/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_agent_state.py index bcf867b43..fac2d1a4e 100644 --- a/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_agent_state.py +++ b/mephisto/abstractions/blueprints/parlai_chat/parlai_chat_agent_state.py @@ -57,15 +57,15 @@ def _load_data(self) -> None: self.messages = state["outputs"]["messages"] self.init_data = state["inputs"] self.final_submission = state["outputs"].get("final_submission") - if 'metadata' in state: + if "metadata" in state: self.metadata = _AgentStateMetadata(**state["metadata"]) - elif 'times' in state: + elif "times" in state: self.metadata = _AgentStateMetadata( - start_time=state['times']['start_time'], - end_time=state['times']['end_time'], + start_time=state["times"]["start_time"], + end_time=state["times"]["end_time"], ) else: - self.meatadata = _AgentStateMetadata() + self.metadata = _AgentStateMetadata() def get_data(self) -> Dict[str, Any]: """Return dict with the messages of this agent"""