Update src/lighteval/models/transformers/transformers_model.py

clefourrier · NathanHB · web-flow · commit e233190e0ac3 · 2024-12-26T12:33:57.000+01:00
Co-authored-by: Nathan Habib &lt;30601243+NathanHB@users.noreply.github.com&gt;
diff --git a/src/lighteval/models/transformers/transformers_model.py b/src/lighteval/models/transformers/transformers_model.py
@@ -1353,7 +1353,8 @@ def __post_init__(self):
         super().__post_init__()
 
         warnings.warn(
-            "Careful, the BaseModel name is deprecated and will be removed, you should use TransformersModel instead!"
+            "Careful, the BaseModel name is deprecated and will be removed, you should use TransformersModel instead!",
+             FutureWarning,
         )
 
 

Original file line number	Diff line number	Diff line change
`@@ -1353,7 +1353,8 @@ def __post_init__(self):`
`1353`	`1353`	`super().__post_init__()`
`1354`	`1354`
`1355`	`1355`	`warnings.warn(`
`1356`		`- "Careful, the BaseModel name is deprecated and will be removed, you should use TransformersModel instead!"`
	`1356`	`+ "Careful, the BaseModel name is deprecated and will be removed, you should use TransformersModel instead!",`
	`1357`	`+ FutureWarning,`
`1357`	`1358`	`)`
`1358`	`1359`
`1359`	`1360`