How do I upload images and videos using WandbLoggerCallback

I thought it would be possible to add either Images or Videos to the episode as a custom_metric or similar during a on_episode_end call. however the custom_metric dict only accepts float values.

The images/videos are created by the environment wrapper and placed in the ‘info’ dict periodically

How do i get this data into the results that are passed to the WandbLogger?

(I think I can do it in the next version due to this: ray/custom_metrics_and_callbacks.py at master · ray-project/ray · GitHub) Is there a way to do this on 1.2.0?

Thanks

Created this PR to try to solve this issue: