__depth.py

def __init__(self, max_depth = 4, depth = 1):
    self.max_depth = max_depth
    self.depth = depth
    self.left = None
    self.right = None
    
def __create_branches(self):
    self.left = DecisionTree(max_depth = self.max_depth, 
                             depth = self.depth + 1)
    self.right = DecisionTree(max_depth = self.max_depth, 
                             depth = self.depth + 1)
    # Rest of code in this method would remain the same.