Fix missing pygame display init, check before.

cooperative_cuisine/pygame_2d_vis/drawing.py

@@ -113,10 +113,10 @@ class Visualizer:
         self.font = pygame.font.SysFont("Arial", 20)
         self.init_get_state_image = False
 
-        # TODO FIX THIS
-        # self.observation_screen = pygame.display.set_mode(
-        #     (100, 100), flags=pygame.HIDDEN
-        # )
+        if not pygame.display.get_init():
+            self.observation_screen = pygame.display.set_mode(
+                (100, 100), flags=pygame.HIDDEN
+            )
 
         self.grid_size = 48