Fix dataset sample

microsoft · Jul 20, 2024 · d72c56a · d72c56a
1 parent f436d5f
commit d72c56a
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 10 deletions.
diff --git a/torchgeo/datasets/cyclone.py b/torchgeo/datasets/cyclone.py
@@ -108,7 +108,8 @@ def __getitem__(self, index: int) -> dict[str, Any]:
 
         if self.transforms is not None:
             sample = self.transforms(sample)
-            sample.update({x: features[x] for x in features if x != 'label'})
+
+        sample.update({x: features[x] for x in features if x != 'label'})
 
         return sample
 

diff --git a/torchgeo/datasets/quakeset.py b/torchgeo/datasets/quakeset.py
@@ -117,7 +117,8 @@ def __getitem__(self, index: int) -> dict[str, Tensor]:
 
         if self.transforms is not None:
             sample = self.transforms(sample)
-            sample['magnitude'] = magnitude
+
+        sample['magnitude'] = magnitude
 
         return sample
 

diff --git a/torchgeo/datasets/skippd.py b/torchgeo/datasets/skippd.py
@@ -149,7 +149,8 @@ def __getitem__(self, index: int) -> dict[str, str | Tensor]:
 
         if self.transforms is not None:
             sample = self.transforms(sample)
-            sample.update({x: features[x] for x in features if x != 'label'})
+
+        sample.update({x: features[x] for x in features if x != 'label'})
 
         return sample
 

diff --git a/torchgeo/datasets/sustainbench_crop_yield.py b/torchgeo/datasets/sustainbench_crop_yield.py
@@ -153,13 +153,10 @@ def __getitem__(self, index: int) -> dict[str, Tensor]:
 
         if self.transforms is not None:
             sample = self.transforms(sample)
-            sample.update(
-                {
-                    x: self.features[index][x]
-                    for x in self.features[index]
-                    if x != 'label'
-                }
-            )
+
+        sample.update(
+            {x: self.features[index][x] for x in self.features[index] if x != 'label'}
+        )
 
         return sample