From 1adf5df031ee33c40304b33a2354d0fd8b037d9a Mon Sep 17 00:00:00 2001 From: Parteek Date: Fri, 31 Jan 2025 18:23:59 +0530 Subject: [PATCH] fix minor issue in e2b (#436) --- src/smolagents/e2b_executor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/smolagents/e2b_executor.py b/src/smolagents/e2b_executor.py index fe87011..38fcb01 100644 --- a/src/smolagents/e2b_executor.py +++ b/src/smolagents/e2b_executor.py @@ -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",