Replace pytext_lib's MaskTransform with new one from torchtext

Summary: Replace instances of `pytext_lib`'s `MaskTransform` with new one from `torchtext` that was merged in #1882 Reviewed By: Nayef211 Differential Revision: D39058074 fbshipit-source-id: f61499d88eec7eccda659279786528bac7edf9d0
pytorch · Aug 30, 2022 · fd6e5cd · fd6e5cd
1 parent 0303323
commit fd6e5cd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchtext/transforms.py b/torchtext/transforms.py
@@ -8,11 +8,11 @@
 from torch import Tensor
 from torch.nn import Module
 from torchtext._torchtext import (
+ BERTEncoder as BERTEncoderPyBind,
  CLIPEncoder as CLIPEncoderPyBind,
  GPT2BPEEncoder as GPT2BPEEncoderPyBind,
- BERTEncoder as BERTEncoderPyBind,
+ RegexTokenizer as RegexTokenizerPybind,
 )
-from torchtext._torchtext import RegexTokenizer as RegexTokenizerPybind
 from torchtext.data.functional import load_sp_model
 from torchtext.utils import get_asset_local_path
 from torchtext.vocab import Vocab