Updated type hint in calculate function

huggingface · Jan 13, 2025 · 155c059 · 155c059
1 parent 730f312
commit 155c059
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/8_agents/notebooks/agents.ipynb b/8_agents/notebooks/agents.ipynb
@@ -85,10 +85,10 @@
             "outputs": [],
             "source": [
                 "from smolagents import CodeAgent, tool\n",
-                "from typing import Union, List\n",
+                "from typing import Union\n",
                 "\n",
                 "@tool\n",
-                "def calculate(operation: str, numbers: List[float]) -> float:\n",
+                "def calculate(operation: str, numbers: object) -> float:\n",
                 "    \"\"\"Performs basic mathematical operations on a list of numbers.\n",
                 "    \n",
                 "    Args:\n",