Update tools.py
Browse files
tools.py
CHANGED
|
@@ -16,6 +16,7 @@ import csv
|
|
| 16 |
import torch.nn.utils.prune as prune
|
| 17 |
from typing import Optional
|
| 18 |
from pydantic import BaseModel, Field
|
|
|
|
| 19 |
|
| 20 |
# Configurer le logging
|
| 21 |
logging.basicConfig(level=logging.INFO)
|
|
|
|
| 16 |
import torch.nn.utils.prune as prune
|
| 17 |
from typing import Optional
|
| 18 |
from pydantic import BaseModel, Field
|
| 19 |
+
from smolagents import Tool
|
| 20 |
|
| 21 |
# Configurer le logging
|
| 22 |
logging.basicConfig(level=logging.INFO)
|