Skip to content

Commit 3f1c603

Browse files
committed
fix: use __import__ to avoid flake8 E402 in test
1 parent 594256d commit 3f1c603

1 file changed

Lines changed: 11 additions & 11 deletions

File tree

tests/test_tools/test_get_flops_grounding.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -8,17 +8,17 @@
88
from mmengine.config import Config
99

1010
# Add tools path so we can import the module directly
11-
sys.path.insert(0, 'tools/analysis_tools') # pragma: no cover
12-
from get_flops_grounding import (
13-
_get_backbone_out_channels, # noqa: E402
14-
_get_feature_strides,
15-
_get_neck_out_channels,
16-
_get_transformer_config,
17-
count_neck_flops,
18-
count_text_encoder_flops,
19-
count_transformer_flops,
20-
format_flops,
21-
format_params)
11+
sys.path.insert(0, 'tools/analysis_tools')
12+
_mod = __import__('get_flops_grounding')
13+
_get_backbone_out_channels = _mod._get_backbone_out_channels
14+
_get_feature_strides = _mod._get_feature_strides
15+
_get_neck_out_channels = _mod._get_neck_out_channels
16+
_get_transformer_config = _mod._get_transformer_config
17+
count_neck_flops = _mod.count_neck_flops
18+
count_text_encoder_flops = _mod.count_text_encoder_flops
19+
count_transformer_flops = _mod.count_transformer_flops
20+
format_flops = _mod.format_flops
21+
format_params = _mod.format_params
2222

2323

2424
class TestFormatFlops(unittest.TestCase):

0 commit comments

Comments
 (0)