[PIR] Fix test cases (#10456)

* [PIR] Fix pir in comments

* [PIR] re-run pre-commit for pir in comments

Author: hanlintang

tests/experimental/autonlp/test_text_classification.py

@@ -24,6 +24,7 @@
 
 from paddlenlp.datasets import load_dataset
 from paddlenlp.experimental.autonlp import AutoTrainerForTextClassification
+from paddlenlp.utils.env import PADDLE_INFERENCE_MODEL_SUFFIX
 from tests.testing_utils import get_tests_dir, slow
 
 finetune_model_candidate = {
@@ -175,13 +176,13 @@ def test_multiclass(self, custom_model_candidate, hp_overrides):
             # test export
             temp_export_path = os.path.join(temp_dir_path, "test_export")
             auto_trainer.export(export_path=temp_export_path)
-            self.assertTrue(os.path.exists(os.path.join(temp_export_path, "model.pdmodel")))
+            self.assertTrue(os.path.exists(os.path.join(temp_export_path, f"model{PADDLE_INFERENCE_MODEL_SUFFIX}")))
             self.assertTrue(os.path.exists(os.path.join(temp_export_path, "taskflow_config.json")))
             self.assertTrue(os.path.exists(os.path.join(temp_export_path, "tokenizer_config.json")))
 
             # test export compress model
             auto_trainer.export(export_path=temp_export_path, compress=True)
-            self.assertTrue(os.path.exists(os.path.join(temp_export_path, "model.pdmodel")))
+            self.assertTrue(os.path.exists(os.path.join(temp_export_path, f"model{PADDLE_INFERENCE_MODEL_SUFFIX}")))
             self.assertTrue(os.path.exists(os.path.join(temp_export_path, "taskflow_config.json")))
             self.assertTrue(os.path.exists(os.path.join(temp_export_path, "tokenizer_config.json")))
 
@@ -400,13 +401,13 @@ def test_default_model_candidate(self, language, hp_overrides):
             # test export
             temp_export_path = os.path.join(temp_dir_path, "test_export")
             auto_trainer.export(export_path=temp_export_path)
-            self.assertTrue(os.path.exists(os.path.join(temp_export_path, "model.pdmodel")))
+            self.assertTrue(os.path.exists(os.path.join(temp_export_path, f"model{PADDLE_INFERENCE_MODEL_SUFFIX}")))
             self.assertTrue(os.path.exists(os.path.join(temp_export_path, "taskflow_config.json")))
             self.assertTrue(os.path.exists(os.path.join(temp_export_path, "tokenizer_config.json")))
 
             # test export compress model
             auto_trainer.export(export_path=temp_export_path, compress=True)
-            self.assertTrue(os.path.exists(os.path.join(temp_export_path, "model.pdmodel")))
+            self.assertTrue(os.path.exists(os.path.join(temp_export_path, f"model{PADDLE_INFERENCE_MODEL_SUFFIX}")))
             self.assertTrue(os.path.exists(os.path.join(temp_export_path, "taskflow_config.json")))
             self.assertTrue(os.path.exists(os.path.join(temp_export_path, "tokenizer_config.json")))
 

tests/transformers/chatglm/test_modeling.py

@@ -19,6 +19,11 @@
 import paddle
 
 from paddlenlp.transformers import ChatGLMConfig, ChatGLMForCausalLM, ChatGLMModel
+
+# from paddlenlp.utils.env import (
+#     PADDLE_INFERENCE_MODEL_SUFFIX,
+#     PADDLE_INFERENCE_WEIGHTS_SUFFIX,
+# )
 from tests.transformers.test_configuration_common import ConfigTester
 from tests.transformers.test_generation_utils import GenerationTesterMixin
 from tests.transformers.test_modeling_common import (
@@ -380,8 +385,8 @@ def test_chatglm_lm_head_model(self):
 #                     ),
 #                 )
 
-#                 model_path = os.path.join(tempdir, "model.pdmodel")
-#                 params_path = os.path.join(tempdir, "model.pdiparams")
+#                 model_path = os.path.join(tempdir, f"model{PADDLE_INFERENCE_MODEL_SUFFIX}")
+#                 params_path = os.path.join(tempdir, f"model{PADDLE_INFERENCE_WEIGHTS_SUFFIX}")
 #                 config = paddle.inference.Config(model_path, params_path)
 
 #                 config.disable_gpu()
@@ -450,8 +455,8 @@ def test_chatglm_lm_head_model(self):
 #                     ),
 #                 )
 
-#                 model_path = os.path.join(tempdir, "model.pdmodel")
-#                 params_path = os.path.join(tempdir, "model.pdiparams")
+#                 model_path = os.path.join(tempdir, f"model{PADDLE_INFERENCE_MODEL_SUFFIX}")
+#                 params_path = os.path.join(tempdir, f"model{PADDLE_INFERENCE_WEIGHTS_SUFFIX}")
 #                 config = paddle.inference.Config(model_path, params_path)
 
 #                 config.disable_gpu()

tests/transformers/test_generation_utils.py

@@ -39,6 +39,11 @@
     PretrainedConfig,
     PretrainedTokenizer,
 )
+
+# from paddlenlp.utils.env import (
+#     PADDLE_INFERENCE_MODEL_SUFFIX,
+#     PADDLE_INFERENCE_WEIGHTS_SUFFIX,
+# )
 from tests.testing_utils import slow
 
 
@@ -1199,8 +1204,8 @@ def test_min_new_tokens(self):
 #                 ),
 #             )
 
-#             model_path = os.path.join(tempdir, "model.pdmodel")
-#             params_path = os.path.join(tempdir, "model.pdiparams")
+#             model_path = os.path.join(tempdir, f"model{PADDLE_INFERENCE_MODEL_SUFFIX}")
+#             params_path = os.path.join(tempdir, f"model{PADDLE_INFERENCE_WEIGHTS_SUFFIX}")
 #             config = paddle.inference.Config(model_path, params_path)
 
 #             config.disable_gpu()
@@ -1265,8 +1270,8 @@ def test_min_new_tokens(self):
 #                 ),
 #             )
 
-#             model_path = os.path.join(tempdir, "model.pdmodel")
-#             params_path = os.path.join(tempdir, "model.pdiparams")
+#             model_path = os.path.join(tempdir, f"model{PADDLE_INFERENCE_MODEL_SUFFIX}")
+#             params_path = os.path.join(tempdir, f"model{PADDLE_INFERENCE_WEIGHTS_SUFFIX}")
 #             config = paddle.inference.Config(model_path, params_path)
 
 #             config.disable_gpu()