diff --git a/test/prototype/test_with_asset.py b/test/prototype/test_with_asset.py index e0ab14f4d7..c5a85086f4 100644 --- a/test/prototype/test_with_asset.py +++ b/test/prototype/test_with_asset.py @@ -1,6 +1,8 @@ import os +import platform import shutil import tempfile +import unittest from functools import partial import torch @@ -135,6 +137,8 @@ def test_vocab_from_file(self) -> None: self.assertEqual(v.get_itos(), expected_itos) self.assertEqual(dict(v.get_stoi()), expected_stoi) + # TODO(Nayef211): remove decorator once https://github.com/pytorch/text/issues/1900 is closed + @unittest.skipIf("CI" in os.environ and platform.system() == "Linux", "Test is known to fail on Linux.") def test_vocab_from_raw_text_file(self) -> None: asset_name = "vocab_raw_text_test.txt" asset_path = get_asset_path(asset_name)