Update traffic_gym.py

Fix AttributeError: module 'PIL' has no attribute 'Image'
Atcold · Jul 28, 2020 · a3e4095 · a3e4095
1 parent a259d28
commit a3e4095
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/traffic_gym.py b/traffic_gym.py
@@ -9,6 +9,7 @@
 # from skimage import measure, transform
 # from matplotlib.image import imsave
 import PIL
+from PIL import Image
 from custom_graphics import draw_dashed_line, draw_text, draw_rect
 from gym import core, spaces
 import os