map labels

requirements/datasets.txt

@@ -6,3 +6,4 @@ pandas[parquet]==2.2.3
 pycocotools==2.0.8
 scikit-image==0.25.0
 scipy==1.15.1
+zstandard==0.23.0

requirements/tests.txt

@@ -2,4 +2,3 @@
 nbmake==1.5.5
 pytest==8.3.4
 pytest-cov==6.0.0
-zstandard==0.23.0

tests/datasets/test_bigearthnet.py

@@ -14,6 +14,8 @@
 
 from torchgeo.datasets import BigEarthNet, BigEarthNetV2, DatasetNotFoundError
 
+pytest.importorskip('zstandard', minversion='0.23')
+
 
 class TestBigEarthNet:
     @pytest.fixture(

torchgeo/datasets/bigearthnet.py

@@ -6,6 +6,8 @@
 import glob
 import json
 import os
+import tarfile
+import tempfile
 from collections.abc import Callable
 from typing import ClassVar
 
@@ -19,17 +21,15 @@
 from matplotlib.figure import Figure
 from rasterio.enums import Resampling
 from torch import Tensor
-import tempfile
-import tarfile
 
 from .errors import DatasetNotFoundError
 from .geo import NonGeoDataset
 from .utils import (
     Path,
     download_url,
     extract_archive,
-    sort_sentinel2_bands,
     lazy_import,
+    sort_sentinel2_bands,
 )
 
 
@@ -683,7 +683,7 @@ class BigEarthNetV2(NonGeoDataset):
 
     def __init__(
         self,
-        root: str = 'data',
+        root: Path = 'data',
         split: str = 'train',
         bands: str = 'all',
         transforms: Callable[[dict[str, Tensor]], dict[str, Tensor]] | None = None,