Update helper_functions.py

This commit is contained in:
Frank Xu 2024-04-15 01:37:45 -04:00 committed by GitHub
parent ce732b26dc
commit dc56401827
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -154,7 +154,7 @@ class RenderHelper(object):
if render_screenshot:
# image observation
img_obs = observation["image"]
image = Image.fromarray(img_obs)
image = Image.fromarray(img_obs) # type:ignore
byte_io = io.BytesIO()
image.save(byte_io, format="PNG")
byte_io.seek(0)