Merge pull request #859 from facebookresearch/parlai-agent-state-fix

Fixing bug in ParlAI agent state from 1.0.2

mephisto/VERSION

@@ -1 +1 @@
-1.0.2
+1.0.3

mephisto/abstractions/blueprints/parlai_chat/parlai_chat_agent_state.py

@@ -5,9 +5,13 @@
 # LICENSE file in the root directory of this source tree.
 
 from typing import List, Optional, Dict, Any, Tuple, TYPE_CHECKING
-from mephisto.abstractions.blueprint import AgentState
+from mephisto.abstractions._subcomponents.agent_state import (
+    AgentState,
+    _AgentStateMetadata,
+)
 import os.path
 import time
+import dataclasses
 
 if TYPE_CHECKING:
     from mephisto.data_model.agent import Agent
@@ -53,17 +57,18 @@ 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 = state["metadata"]
+            self.metadata = _AgentStateMetadata(**state["metadata"])
 
     def get_data(self) -> Dict[str, Any]:
         """Return dict with the messages of this agent"""
+        print(self.metadata)
         return {
             "outputs": {
                 "messages": self.messages,
                 "final_submission": self.final_submission,
             },
             "inputs": self.init_data,
-            "metadata": self.metadata,
+            "metadata": dataclasses.asdict(self.metadata),
         }
 
     def get_parsed_data(self) -> Dict[str, Any]:

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mephisto"
-version = "1.0.2"
+version = "1.0.3"
 description = "Crowdsourcing made simpler."
 license = "MIT"
 authors = [
@@ -91,7 +91,7 @@ pdoc = "^10.0.3"
 pylint = "^2.4"
 pytest = "^5.0"
 pytest-cov = "^2.10.1"
-mypy = "^0.931.0"
+mypy = "^0.971.0"
 
 
 [tool.poetry.scripts]