From 6b7a4dfaa02c2395285b2363c4b7f0cc50b83c86 Mon Sep 17 00:00:00 2001 From: Eric Liang Date: Thu, 5 Nov 2020 22:07:57 -0800 Subject: [PATCH] [rllib] Forgot to pass ioctx to child json readers (#11839) * fix ioctx * fix --- rllib/agents/dreamer/dreamer.py | 2 +- rllib/offline/mixed_input.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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)