Fix type hints

Author: adamjstewart

torchgeo/datasets/sentinel.py

@@ -281,7 +281,7 @@ class Sentinel2(Sentinel):
     """
 
     # https://sentiwiki.copernicus.eu/web/s2-mission
-    all_bands = (
+    all_bands: tuple[str, ...] = (
         'B01',
         'B02',
         'B03',

torchgeo/datasets/ssl4eo.py

@@ -365,6 +365,7 @@ class _Metadata(TypedDict):
         filename: str
         md5: str
         bands: list[str]
+        filename_regex: str
 
     metadata: ClassVar[dict[str, _Metadata]] = {
         's1': {
@@ -467,7 +468,7 @@ def __getitem__(self, index: int) -> dict[str, Any]:
 
         images = []
         bounds = []
-        wavelengths = []
+        wavelengths: list[float] = []
         for subdir in subdirs:
             directory = os.path.join(root, subdir)
             if match := re.match(filename_regex, subdir):
@@ -488,7 +489,7 @@ def __getitem__(self, index: int) -> dict[str, Any]:
                         images.append(torch.from_numpy(image.astype(np.float32)))
                 bounds.append(BoundingBox(minx, maxx, miny, maxy, mint, maxt))
 
-        sample = {
+        sample: dict[str, Any] = {
             'image': torch.cat(images),
             'bounds': bounds,
             'wavelengths': wavelengths,