romanbredehoft-zama commited on
Commit
615cfe4
1 Parent(s): 03475de

Remove unused imports

Browse files
Files changed (2) hide show
  1. development.py +1 -1
  2. utils/model.py +2 -5
development.py CHANGED
@@ -17,7 +17,7 @@ from settings import (
17
  CREDIT_BUREAU_COLUMNS,
18
  )
19
  from utils.client_server_interface import MultiInputsFHEModelDev
20
- from utils.model import MultiInputDecisionTreeClassifier, MultiInputDecisionTreeRegressor
21
  from utils.pre_processing import get_pre_processors
22
 
23
 
 
17
  CREDIT_BUREAU_COLUMNS,
18
  )
19
  from utils.client_server_interface import MultiInputsFHEModelDev
20
+ from utils.model import MultiInputDecisionTreeClassifier
21
  from utils.pre_processing import get_pre_processors
22
 
23
 
utils/model.py CHANGED
@@ -13,7 +13,7 @@ from concrete.ml.common.utils import (
13
  check_there_is_no_p_error_options_in_configuration
14
  )
15
  from concrete.ml.quantization.quantized_module import QuantizedModule, _get_inputset_generator
16
- from concrete.ml.sklearn import DecisionTreeClassifier, DecisionTreeRegressor
17
 
18
  class MultiInputModel:
19
 
@@ -132,7 +132,4 @@ class MultiInputModel:
132
  return compiler
133
 
134
  class MultiInputDecisionTreeClassifier(MultiInputModel, DecisionTreeClassifier):
135
- pass
136
-
137
- class MultiInputDecisionTreeRegressor(MultiInputModel, DecisionTreeRegressor):
138
- pass
 
13
  check_there_is_no_p_error_options_in_configuration
14
  )
15
  from concrete.ml.quantization.quantized_module import QuantizedModule, _get_inputset_generator
16
+ from concrete.ml.sklearn import DecisionTreeClassifier
17
 
18
  class MultiInputModel:
19
 
 
132
  return compiler
133
 
134
  class MultiInputDecisionTreeClassifier(MultiInputModel, DecisionTreeClassifier):
135
+ pass