a644060daa4c22aabedcfd9b891d43f9afa88f97,rllib/utils/experimental_dsl.py,CollectMetrics,__call__,#CollectMetrics#Any#,268

Before Change

    def __call__(self, _):
        metrics = LocalIterator.get_metrics()
        if metrics.parent_metrics:
            raise ValueError("TODO: support nested metrics")
        episodes, self.to_be_collected = collect_episodes(
            self.workers.local_worker(),
            self.workers.remote_workers(),
            self.to_be_collected,

After Change

        info = {}
        for metrics in all_metrics:
            info.update(metrics.info)
            for k, counter in metrics.counters.items():
                counters[k] = counter
            for k, timer in metrics.timers.items():
                timers["{}_time_ms".format(k)] = round(timer.mean * 1000, 3)
                if timer.has_units_processed():
                    timers["{}_throughput".format(k)] = round(

Instances