diff --git a/tests/core/controllers/test_trainer_paddle.py b/tests/core/controllers/test_trainer_paddle.py
index 46feafa5..bc5a590c 100644
--- a/tests/core/controllers/test_trainer_paddle.py
+++ b/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