feat(single-mode): adjust training partner score

NateScarlet · Aug 30, 2021 · 0a80b7f · 0a80b7f
1 parent 6ff8b81
commit 0a80b7f
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/auto_derby/single_mode/training/partner.py b/auto_derby/single_mode/training/partner.py
@@ -211,15 +211,17 @@ def __str__(self):
     def score(self, ctx: Context) -> float:
         if self.type == self.TYPE_OTHER:
             return 2 + self.level
-        ret = mathtools.interpolate(
-            ctx.turn_count(),
-            (
-                (0, 5),
-                (24, 3),
-                (48, 2),
-                (72, 0),
-            ),
-        )
+        ret = 0
+        if 0 <= self.level < 4:
+            ret = mathtools.interpolate(
+                ctx.turn_count(),
+                (
+                    (0, 5),
+                    (24, 3),
+                    (48, 2),
+                    (72, 0),
+                ),
+            )
         if self.has_training:
             ret += 7
         return ret