Skip to content

Commit f7d85a6

Browse files
committed
update test
1 parent c29b148 commit f7d85a6

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

tests/datasets/test_coco.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -494,7 +494,13 @@ def test_make_coco_transforms_forwards_crop_decision(self, aug_config, expected)
494494

495495
from rfdetr.datasets.coco import make_coco_transforms
496496

497-
with patch("rfdetr.datasets.coco._build_train_resize_config", return_value=[]) as mock_build:
497+
# Patch AlbumentationsWrapper.from_config so the test exercises only argument
498+
# forwarding; without it the mocked empty configs trigger unrelated
499+
# "Empty augmentation config provided" warnings.
500+
with (
501+
patch("rfdetr.datasets.coco._build_train_resize_config", return_value=[]) as mock_build,
502+
patch("rfdetr.datasets.coco.AlbumentationsWrapper.from_config", return_value=[]),
503+
):
498504
make_coco_transforms("train", 640, aug_config=aug_config)
499505

500506
assert mock_build.call_args.kwargs["include_crop_branch"] is expected

0 commit comments

Comments
 (0)