From 05ccf62a7dcea8ccfd46906e76b38d6487ec4780 Mon Sep 17 00:00:00 2001 From: Chandan Singh Date: Mon, 11 Mar 2024 20:22:17 -0700 Subject: [PATCH] fix test imports --- tests/gam_multitask_test.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/gam_multitask_test.py b/tests/gam_multitask_test.py index 5d6609c5..db63566a 100644 --- a/tests/gam_multitask_test.py +++ b/tests/gam_multitask_test.py @@ -20,12 +20,9 @@ from sklearn.preprocessing import StandardScaler import imodels -from interpret.glassbox import ExplainableBoostingClassifier, ExplainableBoostingRegressor - from sklearn.base import RegressorMixin, ClassifierMixin import os import os.path -from imodels.algebraic.gam_multitask import MultiTaskGAMRegressor, MultiTaskGAMClassifier path_to_tests = os.path.dirname(os.path.realpath(__file__)) @@ -192,6 +189,8 @@ def compare_models(): if __name__ == '__main__': + from interpret.glassbox import ExplainableBoostingClassifier, ExplainableBoostingRegressor + from imodels.algebraic.gam_multitask import MultiTaskGAMRegressor, MultiTaskGAMClassifier # multi_output_classification() # classification() # single_output_self_supervised()