update

markub3327 · Nov 18, 2023 · 6924bbb · 6924bbb
1 parent 75822ef
commit 6924bbb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rl_toolkit/agents/dueling_dqn/server.py b/rl_toolkit/agents/dueling_dqn/server.py
@@ -140,7 +140,7 @@ def __init__(
                             self._env.observation_space.dtype,
                         ),
                         "action": tf.TensorSpec(
-                            [(frame_stack,) + self._env.action_space.shape],
+                            [frame_stack, *self._env.action_space.shape],
                             self._env.action_space.dtype,
                         ),
                         "ext_reward": tf.TensorSpec([frame_stack, 1], tf.float64),