From fd54303dba73e086c61367d5e8bd0c9be0216088 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Micha=C5=82=20Czeka=C5=84ski?= <michal.cz2202@gmail.com>
Date: Thu, 21 May 2020 16:57:33 +0200
Subject: [PATCH] Change type of branch field in DecisionTree class

Type was Any, now it is DecisionTreeBranch.
---
 src/AI/DecisionTrees/DecisionTree.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/AI/DecisionTrees/DecisionTree.py b/src/AI/DecisionTrees/DecisionTree.py
index 87ed395..98d2013 100644
--- a/src/AI/DecisionTrees/DecisionTree.py
+++ b/src/AI/DecisionTrees/DecisionTree.py
@@ -1,13 +1,14 @@
 from typing import List, Any
 
 from src.AI.DecisionTrees.AttributeDefinition import AttributeDefinition
+from src.AI.DecisionTrees.DecisionTreeBranch import DecisionTreeBranch
 
 
 class DecisionTree(object):
     root: AttributeDefinition
-    branches: List[Any]
+    branches: List[DecisionTreeBranch]
 
     def __init__(self, root):
         self.root = root
         self.branches = []
-        self.branchesNum = 0
\ No newline at end of file
+        self.branchesNum = 0