From f6241302a80f4df7a4cde5227e0692cc7508415c Mon Sep 17 00:00:00 2001 From: Sven Mika Date: Wed, 9 Dec 2020 16:41:13 +0100 Subject: [PATCH] [RLlib] Fix issue 12678: MultiAgentBatch has no attribute `total`. (#12704) --- rllib/env/policy_client.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/rllib/env/policy_client.py b/rllib/env/policy_client.py index 7b6ce4306..232f74f1a 100644 --- a/rllib/env/policy_client.py +++ b/rllib/env/policy_client.py @@ -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(