Fix Unit Test (PaddlePaddle#5188)

* replace by small model

* uie-test

* update

Author: linjieccc

paddlenlp/taskflow/information_extraction.py

@@ -438,12 +438,12 @@ class UIETask(Task):
         },
         "__internal_testing__/tiny-random-uie-x": {
             "model_state": [
-                "https://bj.bcebos.com/paddlenlp/models/community/__internal_testing__/tiny-random-uie-x/model_state.pdparams",
-                "b62bdbfec23ee839c5d0a8a7e6920075",
+                "https://bj.bcebos.com/paddlenlp/models/community/__internal_testing__/tiny-random-uie-x_v1.0/model_state.pdparams",
+                "d9b573b31a82b860b6e5a3005d7b879e",
             ],
             "config": [
-                "https://bj.bcebos.com/paddlenlp/models/community/__internal_testing__/tiny-random-uie-x/config.json",
-                "abeaeacca5316e2cbe779f37d986efb4",
+                "https://bj.bcebos.com/paddlenlp/models/community/__internal_testing__/tiny-random-uie-x_v1.0/config.json",
+                "27d715e680596a69d882056a400d97db",
             ],
             "vocab_file": [
                 "https://bj.bcebos.com/paddlenlp/models/community/__internal_testing__/tiny-random-uie-x/vocab.txt",

paddlenlp/taskflow/zero_shot_text_classification.py

@@ -234,23 +234,23 @@ class ZeroShotTextClassificationTask(Task):
         "__internal_testing__/tiny-random-utc": {
             "model_state": [
                 "https://bj.bcebos.com/paddlenlp/models/community/__internal_testing__/tiny-random-utc/model_state.pdparams",
-                "5ca864e9cc07b54e66f5caa184ddfffb",
+                "d303b59447be690530c35c73f8fd03cd",
             ],
             "config": [
                 "https://bj.bcebos.com/paddlenlp/models/community/__internal_testing__/tiny-random-utc/config.json",
-                "1e8b508d0fbefca16f94c96ef72b3dd3",
+                "3420a6638a7c73c6239eb1d7ca1bc5fe",
             ],
             "vocab_file": [
                 "https://bj.bcebos.com/paddlenlp/models/community/__internal_testing__/tiny-random-utc/vocab.txt",
-                "fd630e130f707f860b0edd148f0445d9",
+                "97eb0ec5a5890c8190e10e251af2e133",
             ],
             "special_tokens_map": [
                 "https://bj.bcebos.com/paddlenlp/models/community/__internal_testing__/tiny-random-utc/special_tokens_map.json",
                 "8b3fb1023167bb4ab9d70708eb05f6ec",
             ],
             "tokenizer_config": [
                 "https://bj.bcebos.com/paddlenlp/models/community/__internal_testing__/tiny-random-utc/tokenizer_config.json",
-                "dcb0f3257830c0eb1f2de47f2d86f89a",
+                "258fc552c15cec90046066ca122899e2",
             ],
         },
     }

tests/taskflow/test_information_extraction.py

@@ -12,9 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import os
+import shutil
 import unittest
 
 from paddlenlp import Taskflow
+from paddlenlp.utils.env import PPNLP_HOME
 
 from ..testing_utils import get_tests_dir
 
@@ -32,6 +35,22 @@ def setUpClass(cls):
             model="__internal_testing__/tiny-random-uie-m",
         )
 
+        tiny_uie_m_path = os.path.join(
+            PPNLP_HOME, "taskflow/information_extraction/__internal_testing__/tiny-random-uie-m/"
+        )
+        tiny_uie_x_path = os.path.join(
+            PPNLP_HOME, "taskflow/information_extraction/__internal_testing__/tiny-random-uie-x/"
+        )
+
+        if not os.path.exists(tiny_uie_x_path):
+            os.mkdir(tiny_uie_x_path)
+
+        for file_to_copy in ["vocab.txt", "sentencepiece.bpe.model"]:
+            shutil.copy(
+                os.path.join(tiny_uie_m_path, file_to_copy),
+                os.path.join(tiny_uie_x_path, file_to_copy),
+            )
+
         cls.uie_x = Taskflow(
             task="information_extraction",
             model="__internal_testing__/tiny-random-uie-x",