Fix bugs in relative keypoint handler

1 files changed, 5 insertions, 4 deletions

diff --git a/solo-tool-project/src/solo_tool/handlers.py b/solo-tool-project/src/solo_tool/handlers.py
index b178b3a..1820e86 100644
--- a/solo-tool-project/src/solo_tool/handlers.py
+++ b/solo-tool-project/src/solo_tool/handlers.py
@@ -56,14 +56,15 @@ def keyPointAbsolute(st: SoloTool, kp: float) -> Callable[[], None]:
 def keyPointRelative(st: SoloTool, delta: int) -> Callable[[], None]:
     from bisect import bisect_right, bisect_left
     def f():
+        l = sorted(set(st.keyPoints + [st.keyPoint]))
         if delta > 0:
-            pivot = bisect_right(st.keyPoints, st.keyPoint) - 1
+            pivot = bisect_right(l, st.keyPoint) - 1
         elif delta < 0:
-            pivot = bisect_left(st.keyPoints, st.keyPoint) - 1
+            pivot = bisect_left(l, st.keyPoint)
         else:
             return
-        new = max(min(pivot + delta, len(st.keyPoints) - 1), 0)
-        st.keyPoint = st.keyPoints[new]
+        new = max(min(pivot + delta, len(l) - 1), 0)
+        st.keyPoint = l[new]
     return f
 
 def rateAbsolute(st: SoloTool, value: float) -> Callable[[], None]: