diff --git a/rllib/agents/dreamer/dreamer.py b/rllib/agents/dreamer/dreamer.py index 4543d1401..6cd45b74b 100644 --- a/rllib/agents/dreamer/dreamer.py +++ b/rllib/agents/dreamer/dreamer.py @@ -199,7 +199,7 @@ class DreamerIteration: metrics = _get_shared_metrics() metrics.info[LEARNER_INFO] = fetches metrics.counters[STEPS_SAMPLED_COUNTER] = self.episode_buffer.timesteps - metrics.counter[STEPS_SAMPLED_COUNTER] *= self.repeat + metrics.counters[STEPS_SAMPLED_COUNTER] *= self.repeat res = collect_metrics(local_worker=self.worker) res["info"] = metrics.info res["info"].update(metrics.counters) diff --git a/rllib/offline/mixed_input.py b/rllib/offline/mixed_input.py index 9f23ba952..69e92304d 100644 --- a/rllib/offline/mixed_input.py +++ b/rllib/offline/mixed_input.py @@ -37,7 +37,7 @@ class MixedInput(InputReader): if k == "sampler": self.choices.append(ioctx.default_sampler_input()) else: - self.choices.append(JsonReader(k)) + self.choices.append(JsonReader(k, ioctx)) self.p.append(v) @override(InputReader)