From a630172df592982b8bb4f43947554c45377796a4 Mon Sep 17 00:00:00 2001
From: Stephen Roller <roller@fb.com>
Date: Fri, 12 Mar 2021 10:05:49 -0500
Subject: [PATCH] Support a rewards field

---
 parlai/utils/misc.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/parlai/utils/misc.py b/parlai/utils/misc.py
index 2a25608e049..a292195bd5b 100644
--- a/parlai/utils/misc.py
+++ b/parlai/utils/misc.py
@@ -659,6 +659,11 @@ def convert(key, value):
             or key == 'text_candidates'
         ):
             return tolist(value)
+        elif key == 'reward':
+            try:
+                return int(value)
+            except ValueError:
+                return float(value)
         elif key == 'episode_done':
             return bool(value)
         else: