From b1f1743487318ff133daf4072d42883a9ac30dd8 Mon Sep 17 00:00:00 2001
From: x54-729 <17307130121@fudan.edu.cn>
Date: Mon, 2 May 2022 10:41:02 +0000
Subject: [PATCH] =?UTF-8?q?test=5FTrainer=5Fpaddle=E6=B7=BB=E5=8A=A0=5FNEE?=
 =?UTF-8?q?D=5FIMPORT=5FPADDLE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/core/controllers/test_trainer_paddle.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

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