diff --git a/tests/python/unittest/test_contrib_text.py b/tests/python/unittest/test_contrib_text.py index 0778e5ace1f7..44d4422c7819 100644 --- a/tests/python/unittest/test_contrib_text.py +++ b/tests/python/unittest/test_contrib_text.py @@ -17,9 +17,6 @@ # coding: utf-8 -from __future__ import absolute_import -from __future__ import print_function - from collections import Counter from common import assertRaises @@ -27,6 +24,8 @@ from mxnet.test_utils import * from mxnet.contrib import text +import pytest + def _get_test_str_of_tokens(token_delim, seq_delim): seq1 = token_delim + token_delim.join(['Life', 'is', 'great', '!']) + token_delim + seq_delim @@ -518,6 +517,7 @@ def test_custom_embedding_with_vocabulary(): ) +@pytest.mark.skip(reason='https://github.com/apache/incubator-mxnet/issues/18282') def test_composite_embedding_with_one_embedding(): embed_root = 'embeddings' embed_name = 'my_embed'