From 26f42b9b02b7ebc8e284dbaa7fc1ebbfa6b8bd3a Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Sat, 19 Mar 2022 10:30:20 -0500 Subject: [PATCH] Fix integration tests on macOS/Windows (#468) --- tests/samplers/test_batch.py | 1 + tests/samplers/test_single.py | 1 + 2 files changed, 2 insertions(+) diff --git a/tests/samplers/test_batch.py b/tests/samplers/test_batch.py index 5c114bb86d7..11971637985 100644 --- a/tests/samplers/test_batch.py +++ b/tests/samplers/test_batch.py @@ -51,6 +51,7 @@ def test_len(self, sampler: CustomBatchGeoSampler) -> None: @pytest.mark.parametrize("num_workers", [0, 1, 2]) def test_dataloader(self, sampler: CustomBatchGeoSampler, num_workers: int) -> None: ds = CustomGeoDataset() + ds.index.insert(0, (0, 100, 200, 300, 400, 500)) dl = DataLoader( ds, batch_sampler=sampler, num_workers=num_workers, collate_fn=stack_samples ) diff --git a/tests/samplers/test_single.py b/tests/samplers/test_single.py index aa13b8b56aa..7221a0a14de 100644 --- a/tests/samplers/test_single.py +++ b/tests/samplers/test_single.py @@ -55,6 +55,7 @@ def test_abstract(self) -> None: @pytest.mark.parametrize("num_workers", [0, 1, 2]) def test_dataloader(self, sampler: CustomGeoSampler, num_workers: int) -> None: ds = CustomGeoDataset() + ds.index.insert(0, (0, 100, 200, 300, 400, 500)) dl = DataLoader( ds, sampler=sampler, num_workers=num_workers, collate_fn=stack_samples )