hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[rllib] Forgot to pass ioctx to child json readers (#11839)

Browse source 
* fix ioctx

* fix
This commit is contained in:
Eric Liang 2020-11-05 22:07:57 -08:00 • committed by GitHub
parent 28e7439cf0
commit 6b7a4dfaa0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
rllib/agents/dreamer/dreamer.py
View file

@ -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)

2
rllib/offline/mixed_input.py
View file

@ -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)