requirements.txt

piece search depth 3

Author: mikhail-vlasenko

AI_main.py

@@ -1,7 +1,6 @@
 from find_landings import all_landings
 import numpy as np
 from direct_keys import *
-from copy import deepcopy
 import time
 from figures import piece_weight, find_figure
 from digit import get_field
@@ -226,8 +225,8 @@ def choose_action_depth2(self, field, piece_idx, next_piece, can_hold):
         return results[0]
 
     def place_piece(self, piece, rotation, x_pos, height, rot_now=0, x_pos_now=3, depth=0):
-        if depth == 2:
-            print('depth 2 reached')
+        if depth == 3:
+            print('depth 3 reached')
             return
         rotate = (rotation - rot_now) % 4
         if rotate < 3:

main.py

@@ -1,10 +1,8 @@
 import time
-from copy import deepcopy
 from digit import get_field
 from figures import type_figure_ext
 from AI_main import AI
 import numpy as np
-import cProfile
 
 
 PLAY_SAFE = False

requirements.txt

@@ -0,0 +1,4 @@
+numpy~=1.17.4
+matplotlib~=3.3.0
+mss~=6.0.0
+keyboard~=0.13.5