From 2a44af25190d82fd3a46cd867b2cd2701e09f012 Mon Sep 17 00:00:00 2001
From: x54-729 <17307130121@fudan.edu.cn>
Date: Mon, 9 May 2022 11:34:55 +0000
Subject: [PATCH] =?UTF-8?q?=E4=B8=BA=20test=5Ftrainer=5Fjittor=20=E6=B7=BB?=
 =?UTF-8?q?=E5=8A=A0=20DummyClass=20=E5=92=8C=20pytest.mark.jittor?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/core/controllers/test_trainer_jittor.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/core/controllers/test_trainer_jittor.py b/tests/core/controllers/test_trainer_jittor.py
index d0eac8cd..30e5e668 100644
--- a/tests/core/controllers/test_trainer_jittor.py
+++ b/tests/core/controllers/test_trainer_jittor.py
@@ -11,6 +11,9 @@ if _NEED_IMPORT_JITTOR:
     import jittor as jt
     from jittor import nn, Module
     from jittor.dataset import Dataset
+else:
+    from fastNLP.core.utils.dummy_class import DummyClass as Module
+    from fastNLP.core.utils.dummy_class import DummyClass as Dataset
 
 
 class JittorNormalModel_Classification(Module):
@@ -68,6 +71,7 @@ class TrainJittorConfig:
 
 @pytest.mark.parametrize("driver,device", [("jittor", None)])
 @pytest.mark.parametrize("callbacks", [[RichCallback(100)]])
+@pytest.mark.jittor
 def test_trainer_jittor(
         driver,
         device,