try fix

bentoml · Dec 23, 2021 · 74794c0 · 74794c0
1 parent 78f6605
commit 74794c0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git 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
@@ -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]])