Enable test

dmlc · Nov 20, 2019 · 74071f7 · 74071f7
1 parent fc2d580
commit 74071f7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/scripts/tests/test_scripts.py b/scripts/tests/test_scripts.py
@@ -135,13 +135,12 @@ def test_sentiment_analysis_textcnn():
                                      '--dropout', '0.5', '--model_mode', 'rand', '--data_name', 'MR'])
     time.sleep(5)
 
-@pytest.mark.skip_master
 @pytest.mark.remote_required
 @pytest.mark.gpu
 @pytest.mark.serial
 @pytest.mark.integration
 @pytest.mark.parametrize('method', ['beam_search', 'sampling'])
-@pytest.mark.parametrize('lmmodel', ['awd_lstm_lm_1150', 'gpt2'])
+@pytest.mark.parametrize('lmmodel', ['awd_lstm_lm_1150', 'gpt2_117m'])
 def test_sampling(method, lmmodel):
     if lmmodel == 'gpt2' and method == 'beam_search':
         return  # unsupported