Fix mypy

taylorhansen · Nov 27, 2023 · cc30002 · cc30002
1 parent 48eb67e
commit cc30002
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/py/serve.py b/src/py/serve.py
@@ -178,7 +178,7 @@ async def server(
                 )
                 q_values = None
             ranked_actions = tf.squeeze(ranked_actions, axis=1)
-            hiddens |= zip(keys, zip(*map(tf.unstack, new_hiddens)))
+            hiddens |= zip(keys, map(list, zip(*map(tf.unstack, new_hiddens))))
 
         ranked_actions = decode_action_rankings(ranked_actions)
         if q_values is not None: