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(