Update helper_functions.py

This commit is contained in:
Frank Xu 2024-04-15 01:39:16 -04:00 committed by GitHub
parent dc56401827
commit b4764781ff
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) # type:ignore
image = Image.fromarray(img_obs) # type:ignore
byte_io = io.BytesIO()
image.save(byte_io, format="PNG")
byte_io.seek(0)