test_Trainer_paddle添加_NEED_IMPORT_PADDLE

tests/core/controllers/test_trainer_paddle.py

@@ -4,9 +4,11 @@ from dataclasses import dataclass
 from fastNLP.core.controllers.trainer import Trainer
 from fastNLP.core.metrics.accuracy import Accuracy
 from fastNLP.core.callbacks.progress_callback import RichCallback
+from fastNLP.envs.imports import _NEED_IMPORT_PADDLE
 
-from paddle.optimizer import Adam
-from paddle.io import DataLoader
+if _NEED_IMPORT_PADDLE:
+    from paddle.optimizer import Adam
+    from paddle.io import DataLoader
 
 
 from tests.helpers.models.paddle_model import PaddleNormalModel_Classification_1