From b3cf630ef02114871fdd61775234c9a7bd37ca23 Mon Sep 17 00:00:00 2001 From: Bhavya Shah Date: Sun, 11 Oct 2026 16:30:34 +0530 Subject: [PATCH] Add types for test_multi_view_collate.py --- pyproject.toml | 1 - tests/data/test_multi_view_collate.py | 8 ++++---- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index ff2f3ce72..8ea2948b7 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -273,7 +273,6 @@ exclude = '''(?x)( tests/cli/test_cli_get_lighty_config.py | tests/cli/test_cli_embed.py | tests/core/test_Core.py | - tests/data/test_multi_view_collate.py | tests/data/test_LightlySubset.py | tests/data/test_LightlyDataset.py | tests/embedding/test_callbacks.py | diff --git a/tests/data/test_multi_view_collate.py b/tests/data/test_multi_view_collate.py index 4dcc2bc5b..e4d8b4fb0 100644 --- a/tests/data/test_multi_view_collate.py +++ b/tests/data/test_multi_view_collate.py @@ -7,16 +7,16 @@ from lightly.data.multi_view_collate import MultiViewCollate -def test_empty_batch(): +def test_empty_batch() -> None: multi_view_collate = MultiViewCollate() - batch = [] + batch: List[Tuple[List[Tensor], int, str]] = [] views, labels, fnames = multi_view_collate(batch) assert len(views) == 0 assert len(labels) == 0 assert len(fnames) == 0 -def test_single_item_batch(): +def test_single_item_batch() -> None: multi_view_collate = MultiViewCollate() img = [torch.randn((3, 224, 224)) for _ in range(5)] label = 1 @@ -29,7 +29,7 @@ def test_single_item_batch(): assert fnames == ["image1.jpg"] -def test_multiple_item_batch(): +def test_multiple_item_batch() -> None: multi_view_collate = MultiViewCollate() img1 = [torch.randn((3, 224, 224)) for _ in range(5)] label1 = 1