mypy fixes

microsoft · Feb 23, 2022 · 3104706 · 3104706
1 parent 447652a
commit 3104706
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/samplers/test_batch.py b/tests/samplers/test_batch.py
@@ -76,7 +76,7 @@ def test_abstract(self, dataset: CustomGeoDataset) -> None:
 
 class TestRandomBatchGeoSampler:
     @pytest.fixture(scope="class")
-    def dataset(self):
+    def dataset(self) -> CustomGeoDataset:
         ds = CustomGeoDataset()
         ds.index.insert(0, (0, 100, 200, 300, 400, 500))
         ds.index.insert(1, (0, 100, 200, 300, 400, 500))