diff --git a/.idea/misc.xml b/.idea/misc.xml
index 715ed69..8eab719 100644
--- a/.idea/misc.xml
+++ b/.idea/misc.xml
@@ -3,5 +3,5 @@
-
+
\ No newline at end of file
diff --git a/.idea/wozek.iml b/.idea/wozek.iml
index b1bd648..004f809 100644
--- a/.idea/wozek.iml
+++ b/.idea/wozek.iml
@@ -4,7 +4,7 @@
-
+
\ No newline at end of file
diff --git a/Raporty/final-evaluation.md b/Raporty/final-evaluation.md
index 1fb6ba8..2eed843 100644
--- a/Raporty/final-evaluation.md
+++ b/Raporty/final-evaluation.md
@@ -7,4 +7,4 @@ Do magazynu przyjerzdza dostawa 20 produktów, które są selekcjonowane przez a
Przy wykorzystaniu drzewa decyzyjnego, Agent rozpoznaje produkty pochodzące z dostawy na podstawie ich cech (kolor, kształt, masa, rozmiar), a następnie nadaje im unikalny kod produktu tworzony na podstawie tych cech.
### Digit recognizer
-Każdemu produktowi przydzielany jest kod produktu i zapisywany w pormacie .png. Recognizer rozpoznaje ciag cyfr, a nastepnie przekazyje go do metody `assigning.py`, która przydziela produkty do odpowiednich półek.
\ No newline at end of file
+Każdemu produktowi przydzielany jest kod produktu i zapisywany w formacie .png. Recognizer rozpoznaje ciag cyfr, a nastepnie przekazyje go do metody `assigning.py`, która przydziela produkty do odpowiednich półek.
\ No newline at end of file
diff --git a/assigning.py b/assigning.py
index 409b926..182e38c 100644
--- a/assigning.py
+++ b/assigning.py
@@ -5,8 +5,8 @@ from codes_recognizer.rocognizer import recognizer
def assigning(product_code_path, shelfs):
c = None
- code = product_code_path[0]
- # code = recognizer(product_code_path)
+ # code = product_code_path[0]
+ code = recognizer(product_code_path[1])
product_color = str(code[0]) + str(code[1])
for shelf in shelfs:
if shelf.color == product_color:
diff --git a/main.py b/main.py
index 8faa61c..89ebeba 100644
--- a/main.py
+++ b/main.py
@@ -116,5 +116,4 @@ def run():
pygame.display.flip()
-
run()
diff --git a/shelf.py b/shelf.py
index 82eede8..5f525cb 100644
--- a/shelf.py
+++ b/shelf.py
@@ -26,14 +26,3 @@ class Shelf:
def get_color(self):
return self.color
- def levels(self):
- r = ''
- str = [0, 0, 0]
- if len(self.level1) > 0:
- str[0] = 1
- if len(self.level2) > 0:
- str[1] = 1
- if len(self.level1) > 0:
- str[2] = 1
- r.join(str)
- return r
\ No newline at end of file