Merge pull request #4 from pkhalaj/fix/typo-with-n_dim

Fix #3 by replacing `self.n_dims` (typo) with `self.n_dim` (correct)
simonpf · Nov 25, 2024 · 1feefac · 1feefac
2 parents 10f0fba + 132b49d
commit 1feefac
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/chimp/data/gridsat.py b/chimp/data/gridsat.py
@@ -206,7 +206,7 @@ def load_sample(
         rel_scale = self.scale / base_scale
 
         if isinstance(crop_size, int):
-            crop_size = (crop_size,) * self.n_dims
+            crop_size = (crop_size,) * self.n_dim
         crop_size = tuple((int(size / rel_scale) for size in crop_size))
 
         if input_file is not None:

diff --git a/chimp/data/input.py b/chimp/data/input.py
@@ -182,7 +182,7 @@ def load_sample(
         rel_scale = self.scale / base_scale
 
         if isinstance(crop_size, int):
-            crop_size = (crop_size,) * self.n_dims
+            crop_size = (crop_size,) * self.n_dim
         crop_size = tuple((int(size / rel_scale) for size in crop_size))
 
         if input_file is not None:

diff --git a/chimp/data/patmosx.py b/chimp/data/patmosx.py
@@ -262,7 +262,7 @@ def load_sample(
         rel_scale = self.scale / base_scale
 
         if isinstance(crop_size, int):
-            crop_size = (crop_size,) * self.n_dims
+            crop_size = (crop_size,) * self.n_dim
         crop_size = tuple((int(size / rel_scale) for size in crop_size))
 
         if input_file is not None:

diff --git a/chimp/data/reference.py b/chimp/data/reference.py
@@ -153,7 +153,7 @@ def load_sample(
         from pytorch_retrieve.tensors.masked_tensor import MaskedTensor
         rel_scale = self.scale / base_scale
         if isinstance(crop_size, int):
-            crop_size = (crop_size,) * self.n_dims
+            crop_size = (crop_size,) * self.n_dim
         crop_size = tuple((int(size / rel_scale) for size in crop_size))
         row_slice, col_slice = scale_slices(slices, rel_scale)