[Release 0.18] Add removeable warnings indicating this is the last release (#2248) (#2257)

* Add removeable warnings indicating this is the last release (#2248)

* make pytorch dep a lower bound (#2256)

* Fix CNNDM dataset tests (#2246)

---------

Co-authored-by: Nicolas Hug <[email protected]>

Author: atalman

setup.py

@@ -66,7 +66,7 @@ def _init_submodule():
 
 pytorch_package_dep = "torch"
 if pytorch_package_version is not None:
-    pytorch_package_dep += "==" + pytorch_package_version
+    pytorch_package_dep += ">=" + pytorch_package_version
 
 
 class clean(distutils.command.clean.clean):

torchtext/__init__.py

@@ -2,15 +2,25 @@
 
 from torch.hub import _get_torch_home
 
+_WARN = True
+_TORCHTEXT_DEPRECATION_MSG = (
+    "\n/!\ IMPORTANT WARNING ABOUT TORCHTEXT STATUS /!\ \n"
+    "Torchtext is deprecated and the last released version will be 0.18 (this one). "
+    "You can silence this warning by calling the following at the beginnign of your scripts: "
+    "`import torchtext; torchtext.disable_torchtext_deprecation_warning()`"
+)
+
+def disable_torchtext_deprecation_warning():
+    global _WARN
+    _WARN = False
+
 # the following import has to happen first in order to load the torchtext C++ library
 from torchtext import _extension  # noqa: F401
 
 _TEXT_BUCKET = "https://download.pytorch.org/models/text/"
 
 _CACHE_DIR = os.path.expanduser(os.path.join(_get_torch_home(), "text"))
 
-from . import data, datasets, prototype, functional, models, nn, transforms, utils, vocab, experimental
-
 try:
     from .version import __version__, git_version  # noqa: F401
 except ImportError:

torchtext/data/__init__.py

@@ -1,3 +1,9 @@
+import warnings
+import torchtext
+if torchtext._WARN:
+    warnings.warn(torchtext._TORCHTEXT_DEPRECATION_MSG)
+
+
 from .functional import (
     custom_replace,
     filter_wikipedia_xml,

torchtext/datasets/__init__.py

@@ -1,3 +1,9 @@
+import warnings
+import torchtext
+if torchtext._WARN:
+    warnings.warn(torchtext._TORCHTEXT_DEPRECATION_MSG)
+
+
 import importlib
 
 from .ag_news import AG_NEWS

torchtext/datasets/cnndm.py

@@ -77,12 +77,21 @@ def _hash_urls(s: tuple):
 
 
 def _get_split_list(source: str, split: str):
+    from torchdata.datapipes.iter import (  # noqa
+        IterableWrapper,
+        OnlineReader,
+    )
     url_dp = IterableWrapper([SPLIT_LIST[source + "_" + split]])
     online_dp = OnlineReader(url_dp)
     return online_dp.readlines().map(fn=_hash_urls)
 
 
 def _load_stories(root: str, source: str, split: str):
+    from torchdata.datapipes.iter import (  # noqa
+        FileOpener,
+        IterableWrapper,
+        GDriveReader,
+    )
     split_list = set(_get_split_list(source, split))
     story_dp = IterableWrapper([URL[source]])
     cache_compressed_dp = story_dp.on_disk_cache(
@@ -135,12 +144,6 @@ def CNNDM(root: str, split: Union[Tuple[str], str]):
         raise ModuleNotFoundError(
             "Package `torchdata` not found. Please install following instructions at https://github.com/pytorch/data"
         )
-    from torchdata.datapipes.iter import (  # noqa
-        FileOpener,
-        IterableWrapper,
-        OnlineReader,
-        GDriveReader,
-    )
 
     cnn_dp = _load_stories(root, "cnn", split)
     dailymail_dp = _load_stories(root, "dailymail", split)

torchtext/experimental/__init__.py

@@ -0,0 +1,4 @@
+import warnings
+import torchtext
+if torchtext._WARN:
+    warnings.warn(torchtext._TORCHTEXT_DEPRECATION_MSG)

torchtext/functional.py

@@ -1,3 +1,8 @@
+import warnings
+import torchtext
+if torchtext._WARN:
+    warnings.warn(torchtext._TORCHTEXT_DEPRECATION_MSG)
+
 from typing import Any, List, Optional
 
 import torch

torchtext/models/__init__.py

@@ -1,2 +1,7 @@
+import warnings
+import torchtext
+if torchtext._WARN:
+    warnings.warn(torchtext._TORCHTEXT_DEPRECATION_MSG)
+
 from .roberta import *  # noqa: F401, F403
 from .t5 import *  # noqa: F401, F403

torchtext/nn/__init__.py

@@ -1 +1,6 @@
+import warnings
+import torchtext
+if torchtext._WARN:
+    warnings.warn(torchtext._TORCHTEXT_DEPRECATION_MSG)
+
 from .modules import *  # noqa: F401,F403

torchtext/prototype/__init__.py

@@ -1,3 +1,8 @@
+import warnings
+import torchtext
+if torchtext._WARN:
+    warnings.warn(torchtext._TORCHTEXT_DEPRECATION_MSG)
+
 from . import transforms
 
 __all__ = ["transforms"]