From 74794c0339b6ae3716236c3cbef995d671a9dff8 Mon Sep 17 00:00:00 2001 From: bojiang Date: Thu, 23 Dec 2021 20:38:02 +0800 Subject: [PATCH] try fix --- bentoml/_internal/frameworks/pytorch_lightning.py | 2 +- .../integration/frameworks/test_pytorch_lightning_impl.py | 7 +++++-- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/bentoml/_internal/frameworks/pytorch_lightning.py b/bentoml/_internal/frameworks/pytorch_lightning.py index dce88f98f24..46b2156461c 100644 --- a/bentoml/_internal/frameworks/pytorch_lightning.py +++ b/bentoml/_internal/frameworks/pytorch_lightning.py @@ -1,7 +1,7 @@ import typing as t from typing import TYPE_CHECKING -import setuptools # type: ignore[reportUnusedImport] +import setuptools # type: ignore[reportUnusedImport] noqa: F401 from simple_di import inject from simple_di import Provide diff --git a/tests/integration/frameworks/test_pytorch_lightning_impl.py b/tests/integration/frameworks/test_pytorch_lightning_impl.py index cb4d798e15d..c101a7d8eb2 100644 --- a/tests/integration/frameworks/test_pytorch_lightning_impl.py +++ b/tests/integration/frameworks/test_pytorch_lightning_impl.py @@ -1,15 +1,18 @@ import math -import torch import pandas as pd import psutil import pytest import torch.nn as nn -import pytorch_lightning as pl +import setuptools # type: ignore[reportUnusedImport] noqa: F401 import bentoml from tests.utils.helpers import assert_have_file_extension +import torch # isort:skip +import pytorch_lightning as pl # isort:skip + + test_df = pd.DataFrame([[5, 4, 3, 2]])