Fix tests

microsoft · Feb 9, 2025 · 39290ab · 39290ab
1 parent c87eea6
commit 39290ab
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/torchgeo/datasets/ssl4eo.py b/torchgeo/datasets/ssl4eo.py
@@ -474,7 +474,12 @@ def __getitem__(self, index: int) -> dict[str, Tensor]:
             directory = os.path.join(root, subdir)
             if match := re.match(filename_regex, subdir):
                 date_str = match.group('date')
-                mint, maxt = disambiguate_timestamp(date_str, Sentinel.date_format)
+                match self.split:
+                    case 's1':
+                        date_format = Sentinel1.date_format
+                    case 's2c' | 's2a':
+                        date_format = Sentinel2.date_format
+                mint, maxt = disambiguate_timestamp(date_str, date_format)
                 for band in self.bands:
                     match self.split:
                         case 's1':