From a11be71b1bfd630fa772d427b670b55045aeed0b Mon Sep 17 00:00:00 2001 From: AHReccese Date: Thu, 21 Sep 2023 02:58:43 +0330 Subject: [PATCH] add extra tree's tests to decision tree tester --- tests/test_decision_trees/test_decision_trees.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/test_decision_trees/test_decision_trees.py b/tests/test_decision_trees/test_decision_trees.py index 0d0bf0bd..c53387de 100644 --- a/tests/test_decision_trees/test_decision_trees.py +++ b/tests/test_decision_trees/test_decision_trees.py @@ -3,9 +3,12 @@ from decision_tree.decision_tree_regression import decision_tree_regression from decision_tree.decision_tree_classification import decision_tree_classification +from extra_tree.extra_tree_regression import extra_tree_regression +from extra_tree.extra_tree_classification import extra_tree_classification + DECISION_TREES = { "DECISION_TREE": [decision_tree_regression, decision_tree_classification], - + "EXTRA TREE": [extra_tree_regression, extra_tree_classification], } @pytest.fixture(scope="session", autouse=True)