Client support for non-increment time controls

kiudee · Feb 2, 2020 · 7155cf1 · 7155cf1
1 parent cdf7660
commit 7155cf1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/tune/db_workers/tuning_client.py b/tune/db_workers/tuning_client.py
@@ -123,7 +123,10 @@ def adjust_time_control(self, time_control, lc0_nodes, sf_nodes):
         tc_sf = parse_timecontrol(time_control.engine2)
         tc_lc0 = [x * lc0_ratio for x in tc_lc0]
         tc_sf = [x * sf_ratio for x in tc_sf]
-        # TODO: support non-increment time-control
+        if len(tc_lc0) == 1:
+            return TimeControl(
+                engine1=f"{tc_lc0[0]}", engine2=f"{tc_sf[0]}"
+            )
         return TimeControl(
             engine1=f"{tc_lc0[0]}+{tc_lc0[1]}", engine2=f"{tc_sf[0]}+{tc_sf[1]}"
         )

diff --git a/tune/db_workers/utils.py b/tune/db_workers/utils.py
@@ -1,9 +1,10 @@
 from collections import namedtuple
-import numpy as np
 
 MatchResult = namedtuple("MatchResult", ["wins", "losses", "draws"])
 TimeControl = namedtuple("TimeControl", ["engine1", "engine2"])
 
 
 def parse_timecontrol(tc_string):
-    return tuple([float(x) for x in tc_string.split("+")])
+    if "+" in tc_string:
+        return tuple([float(x) for x in tc_string.split("+")])
+    return (float(tc_string),)