Merge pull request #861 from chiehminwei/patch-1

Make `self.metadata` backward-compatible
facebookresearch · Jul 26, 2022 · 55f7937 · 55f7937
2 parents d000d70 + 57ffefd
commit 55f7937
Showing 1 changed file with 9 additions and 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
@@ -57,7 +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")
-            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.metadata = _AgentStateMetadata()
 
     def get_data(self) -> Dict[str, Any]:
         """Return dict with the messages of this agent"""