simple question regarding stag-hunt rendering

Hello,

First, thank you for sharing your code and research, it is really helpful.

During the research, I become curious about how the agents act during training on stag-hunt environment.
But it seems the visualization part on stag-hunt in the uploaded code is not implemented.
Hence, I was wondering if there is any update on rendering part? If so, could you share the update? 