Including logger metadata in the TFDS backend.

google-deepmind / envlogger

A tool for recording RL trajectories.

Apache License 2.0

93 stars 13 forks source link

Including logger metadata in the TFDS backend. #6

Closed ethanluoyc closed 1 year ago

ethanluoyc commented 1 year ago

Hi,

It looks like the TFDS backend does not support recording metadata from envlogger. It would be very useful for having this feature. I think there is a metadata key tfds DatasetConfig that we might be able to use for that.

I am happy to investigate this and create a PR.

kenjitoyama commented 1 year ago

Hello @ethanluoyc, did you try store_ds_metadata? It does end up in TFDS and it can be retrieved later (please see this unit test).

ethanluoyc commented 1 year ago

I didin't see that argument! Thanks for letting me know! Sorry for not noticing this. Closing.