fix(nyz): fix metadata name bug

opendilab · Apr 26, 2023 · d13bcd1 · d13bcd1
1 parent e4d5755
commit d13bcd1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lzero/middleware/collector.py b/lzero/middleware/collector.py
@@ -214,7 +214,7 @@ def __call__(self, ctx):
                     {
                         'priorities': self._trajectory_pool[i][1],
                         'done': self._trajectory_pool[i][2],
-                        '_unroll_plus_td_steps': self._unroll_plus_td_steps
+                        'unroll_plus_td_steps': self._unroll_plus_td_steps
                     } for i in range(L)
                 ]