Ignore torch warning in tests

dottxt-ai · Feb 2, 2024 · 9a2f8de · 9a2f8de
1 parent 5ef6eb0
commit 9a2f8de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -91,7 +91,7 @@ filterwarnings = [
     "ignore::numba.core.errors.NumbaPendingDeprecationWarning",
     "ignore::pydantic.warnings.PydanticDeprecatedSince20",
     "ignore::FutureWarning:transformers.*",
-    "ignore::UserWarning:torch.cuda.*"
+    "ignore::UserWarning",
 ]
 
 [tool.mypy]