diff --git a/test/espnet2/tasks/test_abs_task.py b/test/espnet2/tasks/test_abs_task.py index 7a9297f78e2..6b36d3b51d1 100644 --- a/test/espnet2/tasks/test_abs_task.py +++ b/test/espnet2/tasks/test_abs_task.py @@ -8,7 +8,7 @@ from espnet2.train.collate_fn import CommonCollateFn -class TestModel(AbsESPnetModel): +class DummyModel(AbsESPnetModel): def __init__(self): super().__init__() self.layer1 = torch.nn.Linear(1, 1) @@ -60,7 +60,7 @@ def optional_data_names(cls, train=True, inference=False): @classmethod def build_model(cls, args): - model = TestModel() + model = DummyModel() return model @classmethod