fix codecov

microsoft · Feb 13, 2024 · 44fecda · 44fecda
1 parent 0b5e99a
commit 44fecda
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 11 deletions.
diff --git a/docs/api/datamodules.rst b/docs/api/datamodules.rst
@@ -6,6 +6,11 @@ torchgeo.datamodules
 Geospatial DataModules
 ----------------------
 
+AgriFieldNet
+^^^^^^^^^^^^
+
+.. autoclass:: AgriFieldNetDataModule
+
 Chesapeake Land Cover
 ^^^^^^^^^^^^^^^^^^^^^
 

diff --git a/tests/conf/agrifieldnet.yaml b/tests/conf/agrifieldnet.yaml
@@ -15,14 +15,3 @@ data:
     patch_size: 16
   dict_kwargs:
     paths: "tests/data/agrifieldnet"
-
-
-data:
-  class_path: L7IrishDataModule
-  init_args:
-    batch_size: 1
-    patch_size: 32
-    length: 5
-  dict_kwargs:
-    paths: "tests/data/l7irish"
-    download: true
diff --git a/tests/trainers/test_segmentation.py b/tests/trainers/test_segmentation.py
@@ -55,6 +55,7 @@ class TestSemanticSegmentationTask:
     @pytest.mark.parametrize(
         "name",
         [
+            "agrifieldnet",
             "chabud",
             "chesapeake_cvpr_5",
             "chesapeake_cvpr_7",