precommit

fzi-forschungszentrum-informatik · Aug 8, 2023 · b949537 · b949537
1 parent 7773ea2
commit b949537
Show file tree

Hide file tree

Showing 6 changed files with 31 additions and 12,081 deletions.
diff --git a/TSInterpret/InterpretabilityModels/counterfactual/COMTE/Optimization.py b/TSInterpret/InterpretabilityModels/counterfactual/COMTE/Optimization.py
@@ -1,11 +1,9 @@
 import logging
 import multiprocessing
 import numbers
-import sys
 from typing import Tuple
 import numpy as np
 import pandas as pd
-import six
 from sklearn.neighbors import KDTree
 from skopt import gbrt_minimize, gp_minimize
 from TSInterpret.InterpretabilityModels.counterfactual.COMTE.Problem import (
@@ -16,10 +14,6 @@
     random_hill_climb,
 )
 
-# sys.modules["sklearn.externals.six"] = six
-
-# import mlrose
-
 
 class BaseExplanation:
     def __init__(

diff --git a/...unterfactual/COMTE/Optmization_helpers.py → ...nterfactual/COMTE/Optimization_helpers.py b/...unterfactual/COMTE/Optmization_helpers.py → ...nterfactual/COMTE/Optimization_helpers.py
diff --git a/TSInterpret/InterpretabilityModels/counterfactual/COMTE/__init__.py b/TSInterpret/InterpretabilityModels/counterfactual/COMTE/__init__.py
@@ -1,3 +1,3 @@
-from . import Optimization, Problem, Optmization_helpers
+from . import Optimization, Problem, Optimization_helpers
 
 __all__ = ["Optimization", "Problem", "Optimization_helpers"]
diff --git a/TSInterpret/InterpretabilityModels/counterfactual/COMTECF.py b/TSInterpret/InterpretabilityModels/counterfactual/COMTECF.py
@@ -1,16 +1,13 @@
-import logging
-import multiprocessing
-import numbers
-import sys
 from typing import Tuple
 import numpy as np
-import pandas as pd
 from TSInterpret.InterpretabilityModels.counterfactual.CF import CF
 from TSInterpret.Models.PyTorchModel import PyTorchModel
 from TSInterpret.Models.SklearnModel import SklearnModel
 from TSInterpret.Models.TensorflowModel import TensorFlowModel
-from TSInterpret.InterpretabilityModels.counterfactual.COMTE.Optimization import *
-from TSInterpret.InterpretabilityModels.counterfactual.COMTE.Problem import *
+from TSInterpret.InterpretabilityModels.counterfactual.COMTE.Optimization import (
+    BruteForceSearch,
+    OptimizedSearch,
+)
 
 
 class COMTECF(CF):