💚 Fixing CI Build.

BrikerMan · May 25, 2019 · 363bf54 · 363bf54
1 parent 3e56ecb
commit 363bf54
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/tests/test_classification.py b/tests/test_classification.py
@@ -19,14 +19,15 @@
 
 valid_x, valid_y = SMP2018ECDTCorpus.load_data('valid')
 
+bert_path = get_file('bert_sample_model',
+                     "https://storage.googleapis.com/kashgari/bert_sample_model.tar.bz2",
+                     cache_dir=DATA_PATH,
+                     untar=True)
+
 sample_w2v_path = get_file('sample_w2v.txt',
                            "https://storage.googleapis.com/kashgari/sample_w2v.txt",
                            cache_dir=DATA_PATH)
 
-bert_path = get_file('bert_sample_model',
-                     "https://storage.googleapis.com/kashgari/bert_sample_model.tar.bz2",
-                     cache_dir=DATA_PATH)
-
 w2v_embedding = WordEmbedding(sample_w2v_path, task=kashgari.CLASSIFICATION)
 w2v_embedding_variable_len = WordEmbedding(sample_w2v_path, task=kashgari.CLASSIFICATION, sequence_length='variable')
 

diff --git a/tests/test_embeddings.py b/tests/test_embeddings.py
@@ -128,7 +128,8 @@ def setUpClass(cls):
         cls.embedding_class = BERTEmbedding
         bert_path = get_file('bert_sample_model',
                              "https://storage.googleapis.com/kashgari/bert_sample_model.tar.bz2",
-                             cache_dir=DATA_PATH)
+                             cache_dir=DATA_PATH,
+                             untar=True)
         cls.config = {
             'bert_path': bert_path
         }