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]])