Simplify multiple isinstance

huggingface · Mar 6, 2025 · 7b0b494 · 7b0b494
1 parent e7a73b6
commit 7b0b494
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/src/smolagents/local_python_executor.py b/src/smolagents/local_python_executor.py
@@ -640,13 +640,7 @@ def evaluate_call(
     custom_tools: Dict[str, Callable],
     authorized_imports: List[str],
 ) -> Any:
-    if not (
-        isinstance(call.func, ast.Call)
-        or isinstance(call.func, ast.Lambda)
-        or isinstance(call.func, ast.Attribute)
-        or isinstance(call.func, ast.Name)
-        or isinstance(call.func, ast.Subscript)
-    ):
+    if not isinstance(call.func, (ast.Call, ast.Lambda, ast.Attribute, ast.Name, ast.Subscript)):
         raise InterpreterError(f"This is not a correct function: {call.func}).")
 
     func, func_name = None, None