fix minor issue in e2b (#436)
This commit is contained in:
parent
b453d63f33
commit
1adf5df031
|
@ -137,7 +137,7 @@ locals().update({key: value for key, value in pickle_dict.items()})
|
|||
if getattr(result, attribute_name) is not None:
|
||||
image_output = getattr(result, attribute_name)
|
||||
decoded_bytes = base64.b64decode(image_output.encode("utf-8"))
|
||||
return Image.open(BytesIO(decoded_bytes)), execution_logs
|
||||
return Image.open(BytesIO(decoded_bytes)), execution_logs, self.final_answer
|
||||
for attribute_name in [
|
||||
"chart",
|
||||
"data",
|
||||
|
|
Loading…
Reference in New Issue