Midterm 2023a

irlc/utils/player_wrapper.py

@@ -76,6 +76,11 @@ async def _webassembly_interactive(env, agent, autoplay=False):
     from pygame import gfxdraw
     import numpy as np
 
+    if not hasattr(agent, 'reset'):
+        from irlc.lectures.lec10.utils import agent_reset
+        agent.reset = lambda self: agent_reset(self)
+
+
     def aapolygon(surface, points, color):
         pygame.draw.polygon(surface, color, points)