fix custom branching bug

Destaq · May 5, 2020 · c7ff8ae · c7ff8ae
1 parent 771fbdc
commit c7ff8ae
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/chart.py b/chart.py
@@ -13,10 +13,10 @@
 
 
 def form_values(gammme, depth, fragmentation_percentage, should_defragment, custom_branching):
-    lst, ratios, kick_depth = new_parser.parse_games(gammme, depth, custom_branching) #whether or not to implement custom branching
     """Create parent, id, labels, and values """
+    lst, ratios, kick_depth = new_parser.parse_games(gammme, depth, custom_branching) #whether or not to implement custom branching
     firstx = [
-        lst[i][:depth] for i in range(len(lst))
+        lst[i][:depth+kick_depth] for i in range(len(lst))
     ]  # probably unneeded but for safety's sake...
 
     all_level_moves, exclude_first_moves = [], []  # for parent/labels later
@@ -26,7 +26,7 @@ def form_values(gammme, depth, fragmentation_percentage, should_defragment, cust
 
     percentage_holder, firstmove = [], []
 
-    while counter < depth:
+    while counter < depth + kick_depth:
         counter += 1
         othermove = list(
             Counter([tuple(firstx[i][kick_depth:counter]) for i in range(len(lst))]).items()

diff --git a/new_parser.py b/new_parser.py
@@ -1,5 +1,4 @@
 import pgn
-import sys
 
 
 def parse_games(database, depth, custom_branching):