google-deepmind/envlogger

Including logger metadata in the TFDS backend.

ethanluoyc opened this issue · 2 comments

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.

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

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