From dd25c7a145b940e008f6860b2f56575e3c68954f Mon Sep 17 00:00:00 2001
From: Vadzim Valchkovich <vol4k.w@gmil.com>
Date: Fri, 26 May 2023 03:26:14 +0200
Subject: [PATCH] tree_increase and alt_dataset merge

---
 src/obj/Waiter.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/src/obj/Waiter.py b/src/obj/Waiter.py
index 2090958..c80c4dc 100644
--- a/src/obj/Waiter.py
+++ b/src/obj/Waiter.py
@@ -18,13 +18,13 @@ class Waiter(Object):
         self.orientation = copy.copy(state.orientation)
         self.basket = copy.copy(state.basket)
         self.battery -= state.cost
+        # self.calcTree()
         return state
 
     def dish_in_basket(self, table) -> bool:
         return table in self.basket
-        self.calcTree()
-        return state
 
+    '''    
     def calcTree(self):
         from sklearn import tree
         import pandas as pd  # for manipulating the csv data
@@ -101,10 +101,8 @@ class Waiter(Object):
         else:
             result = "NO"
         print("Predicted Label:", result)
-
-    def dampState(self):
-        self.prev_position = copy.deepcopy(self.position)
-        self.prev_orientation = copy.copy(self.orientation)
+        
+    '''
 
     def basket_is_full(self) -> bool:
         return self.basket_size == 0