Hide warning [release]

datadreamer-dev · Apr 25, 2024 · 7c4ab24 · 7c4ab24
1 parent 46cef5d
commit 7c4ab24
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "DataDreamer"
-version = "0.30.0"
+version = "0.31.0"
 description = "Prompt. Generate Synthetic Data. Train & Align Models."
 license = "MIT"
 authors= [

diff --git a/src/utils/import_utils.py b/src/utils/import_utils.py
@@ -60,7 +60,7 @@ def ignore_training_warnings():
         warnings.filterwarnings(
             "ignore",
             category=UserWarning,
-            message=".*please pass in use_reentrant.*",
+            message=".*use_reentrant.*",
             module="torch.utils.checkpoint",
         )
         warnings.filterwarnings(