From 3d74557e66cff2658c1b0b34cffa83e1ca3a5b62 Mon Sep 17 00:00:00 2001
From: stypoumic <stypoumic@gmail.com>
Date: Fri, 13 Sep 2024 10:58:11 +0000
Subject: [PATCH] Sort chips based on hit_id

---
 torchgeo/samplers/single.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/torchgeo/samplers/single.py b/torchgeo/samplers/single.py
index 5b7a5fad40..f8cd72b2fa 100644
--- a/torchgeo/samplers/single.py
+++ b/torchgeo/samplers/single.py
@@ -240,7 +240,12 @@ def __iter__(self) -> Iterator[BoundingBox]:
         Returns:
             (minx, maxx, miny, maxy, mint, maxt) coordinates to index a dataset
         """
+        # sort chips based on hit_id
+        self.chips = self.chips.sort_values(by=['hit_id'])
+
         for _, chip in self.chips.iterrows():
+            print("------------------------------------")
+            print("Chip FID: {}".format(chip["fid"]))
             yield BoundingBox(
                 chip.minx, chip.maxx, chip.miny, chip.maxy, chip.mint, chip.maxt
             )