🔀 [Merge] branch 'SETUP' into MODELv7

tests/test_tools/test_data_loader.py

@@ -1,12 +1,10 @@
 import sys
 from pathlib import Path
 
-import pytest
-
 project_root = Path(__file__).resolve().parent.parent.parent
 sys.path.append(str(project_root))
 
-from yolo.config.config import Config, TrainConfig
+from yolo.config.config import Config
 from yolo.tools.data_loader import StreamDataLoader, YoloDataLoader, create_dataloader
 
 

yolo/model/module.py

@@ -7,7 +7,6 @@ from loguru import logger
 from torch import Tensor, nn
 from torch.nn.common_types import _size_2_t
 
-from yolo.utils.bounding_box_utils import generate_anchors
 from yolo.utils.module_utils import auto_pad, create_activation_function, round_up