aligned_sequence: tolerate index out of bounds

cfe-lab · Sep 22, 2023 · 43efacd · 43efacd
1 parent 3f90930
commit 43efacd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/util/aligned_sequence.py b/util/aligned_sequence.py
@@ -47,7 +47,11 @@ def map_index(self, index):
         if not isinstance(index, int):
             raise TypeError(f"Expected integer as index", index)
 
-        return self.get_coordinates_mapping()[index]
+        mapping = self.get_coordinates_mapping()
+        if index < len(mapping):
+            return mapping[index]
+        else:
+            return mapping[-1]
 
 
     def index(self, index):