EuroCrops: handle Nones in get_label (#2499)

* handle Nones in get_label

* make ruff hapy

* unit test for the win

* remove print statement

tests/datasets/test_eurocrops.py

@@ -82,6 +82,11 @@ def test_invalid_query(self, dataset: EuroCrops) -> None:
         ):
             dataset[query]
 
+    def test_get_label_with_none_hcat_code(self, dataset: EuroCrops) -> None:
+        mock_feature = {'properties': {dataset.label_name: None}}
+        label = dataset.get_label(mock_feature)
+        assert label == 0, "Expected label to be 0 when 'EC_hcat_c' is None."
+
     def test_integrity_error(self, dataset: EuroCrops) -> None:
         dataset.zenodo_files = (('AA.zip', 'invalid'),)
         assert not dataset._check_integrity()

torchgeo/datasets/eurocrops.py

@@ -204,6 +204,9 @@ def get_label(self, feature: 'fiona.model.Feature') -> int:
         # We go up the class hierarchy until there is a match.
         # (Parent code is computed by replacing rightmost non-0 character with 0.)
         hcat_code = feature['properties'][self.label_name]
+        if hcat_code is None:
+            return 0
+
         while True:
             if hcat_code in self.class_map:
                 return self.class_map[hcat_code]