From cc300024aa9a0944e3d218134b7f50bd4cf141d1 Mon Sep 17 00:00:00 2001 From: taylorhansen Date: Sun, 26 Nov 2023 18:57:11 -0800 Subject: [PATCH] Fix mypy --- src/py/serve.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/py/serve.py b/src/py/serve.py index 0b04371b..70baff9e 100644 --- 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: