Generate: deprecate old src imports (#28607)

huggingface · Jan 27, 2024 · 03cc177 · 03cc177
1 parent a28a769
commit 03cc177
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/transformers/generation_flax_utils.py b/src/transformers/generation_flax_utils.py
@@ -23,6 +23,6 @@ class FlaxGenerationMixin(FlaxGenerationMixin):
     # warning at import time
     warnings.warn(
         "Importing `FlaxGenerationMixin` from `src/transformers/generation_flax_utils.py` is deprecated and will "
-        "be removed in Transformers v5. Import as `from transformers import FlaxGenerationMixin` instead.",
+        "be removed in Transformers v4.40. Import as `from transformers import FlaxGenerationMixin` instead.",
         FutureWarning,
     )
diff --git a/src/transformers/generation_tf_utils.py b/src/transformers/generation_tf_utils.py
@@ -23,6 +23,6 @@ class TFGenerationMixin(TFGenerationMixin):
     # warning at import time
     warnings.warn(
         "Importing `TFGenerationMixin` from `src/transformers/generation_tf_utils.py` is deprecated and will "
-        "be removed in Transformers v5. Import as `from transformers import TFGenerationMixin` instead.",
+        "be removed in Transformers v4.40. Import as `from transformers import TFGenerationMixin` instead.",
         FutureWarning,
     )
diff --git a/src/transformers/generation_utils.py b/src/transformers/generation_utils.py
@@ -23,6 +23,6 @@ class GenerationMixin(GenerationMixin):
     # warning at import time
     warnings.warn(
         "Importing `GenerationMixin` from `src/transformers/generation_utils.py` is deprecated and will "
-        "be removed in Transformers v5. Import as `from transformers import GenerationMixin` instead.",
+        "be removed in Transformers v4.40. Import as `from transformers import GenerationMixin` instead.",
         FutureWarning,
     )