fix test imports

csinva · Mar 12, 2024 · 05ccf62 · 05ccf62
1 parent d0cc967
commit 05ccf62
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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()