Działający podmoduł neuro

Marcin.py

@@ -75,13 +75,12 @@ class main():
         elif self.mode_value in [6]:
             return 10
 
-    def main_collective(self,poz):
+    def main_collective(self,poz = None):
+        if poz is None:
+            poz = self.traktor.get_poz()
         self.pole = self.ui.field_images[self.field.get_value(poz)]
         self.img = pygame.surfarray.array3d(self.pole)
         self.img = self.img.transpose([1, 0, 2])
         self.img = cv2.cvtColor(self.img, cv2.COLOR_RGB2BGR)
-        self.reco = self.mode(self.recognition(self.img))
-        if self.reco == 10:
-            print("Nic nie trzeba robić")
-        else:
-            self.traktor.set_mode(self.reco)
+        self.reco = self.recognition(self.img)
+        return self.reco

collective.py

@@ -11,7 +11,7 @@ class main():
         self.path = path
 
     def main(self):
-        # self.marcin.main_collective(pozycja)  To podaje jakie jest pole jest na danym miejscu
+        print(self.marcin.main_collective())
         pass
 
     # 1. sprawdza czy opłaca się pracować