[RLlib] Fix issue 12678: MultiAgentBatch has no attribute total. (#12704)

2026-07-28 11:25:04 +08:00 · 2020-12-09 16:41:13 +01:00
parent 3ce9286977
commit f6241302a8
1 changed files with 2 additions and 1 deletions
@@ -269,7 +269,8 @@ class _LocalInferenceThread(threading.Thread):
                if isinstance(samples, MultiAgentBatch):
                    logger.info(
                        "Sending batch of {} env steps ({} agent steps) to "
-                        "server.".format(samples.count, samples.total()))
+                        "server.".format(samples.env_steps(),
+                                         samples.agent_steps()))
                else:
                    logger.info(
                        "Sending batch of {} steps back to server.".format(