From c0c8789fe6aa6cd0d2d70d8c3a3b8906e7424859 Mon Sep 17 00:00:00 2001 From: Ethan Harris Date: Mon, 6 Dec 2021 23:26:21 +0000 Subject: [PATCH] fix --- tests/text/classification/test_data_model_integration.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/text/classification/test_data_model_integration.py b/tests/text/classification/test_data_model_integration.py index f7244d5da8..0221cdec2d 100644 --- a/tests/text/classification/test_data_model_integration.py +++ b/tests/text/classification/test_data_model_integration.py @@ -15,8 +15,8 @@ from pathlib import Path import pytest -from pytorch_lightning import Trainer +from flash.core.trainer import Trainer from flash.text import TextClassificationData, TextClassifier from tests.helpers.utils import _TEXT_TESTING