tree_increase and alt_dataset merge

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