KeyArgo
/
smolagents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix output type sanitization

This commit is contained in:
Aymeric 2025-01-10 22:30:29 +01:00
parent 6743d01ed5
commit 82a2fe5bb4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/smolagents/types.py
View File

@ -277,8 +277,8 @@ def handle_agent_output_types(output, output_type=None):
# If the class does not have defined output, then we map according to the type # If the class does not have defined output, then we map according to the type
for _k, _v in INSTANCE_TYPE_MAPPING.items(): for _k, _v in INSTANCE_TYPE_MAPPING.items():
if isinstance(output, _k): if isinstance(output, _k):
if not isinstance( if (
output, object _k is not object
): # avoid converting to audio if torch is not installed ): # avoid converting to audio if torch is not installed
return _v(output) return _v(output)
return output return output