✅ [Fix] expired test of data_augmentation

examples/example_inference.py

@@ -28,7 +28,7 @@ def main(cfg: Config):
     model = get_model(cfg).to(device)
 
     tester = ModelTester(cfg, model, save_path, device)
-    tester.solve(dataloader, cfg.task.epoch)
+    tester.solve(dataloader)
 
 
 if __name__ == "__main__":

tests/test_model/test_yolo.py

@@ -19,6 +19,7 @@ def test_build_model():
         cfg = compose(config_name=config_name)
 
         OmegaConf.set_struct(cfg.model, False)
+        cfg.weight = None
         model = YOLO(cfg.model, 80)
         assert len(model.model) == 38
 
@@ -26,6 +27,7 @@ def test_build_model():
 def test_get_model():
     with initialize(config_path=config_path, version_base=None):
         cfg = compose(config_name=config_name)
+        cfg.weight = None
         model = get_model(cfg)
         assert isinstance(model, YOLO)
 

tests/test_utils/test_dataaugment.py

@@ -39,7 +39,7 @@ def test_compose():
         return image, boxes
 
     compose = AugmentationComposer([mock_transform, mock_transform])
-    img = Image.new("RGB", (10, 10), color="blue")
+    img = Image.new("RGB", (640, 640), color="blue")
     boxes = torch.tensor([[0, 0.2, 0.2, 0.8, 0.8]])
 
     transformed_img, transformed_boxes = compose(img, boxes)