Skip to content

Commit 40213c1

Browse files
committed
Remove custom models and plugins handling
1 parent 034e5cf commit 40213c1

File tree

1 file changed

+1
-9
lines changed

1 file changed

+1
-9
lines changed

dataikuapi/dss/ml.py

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -281,12 +281,6 @@ def get_enabled_algorithm_names(self):
281281
"""
282282
algos = self.__class__.algorithm_remap
283283
algo_names = [algo_name for algo_name in algos.keys() if self.mltask_settings["modeling"][algos[algo_name].algorithm_name.lower()]["enabled"]]
284-
if any(custom_mllib["enabled"] for custom_mllib in self.mltask_settings["modeling"]["custom_mllib"]):
285-
algo_names.append("CUSTOM_MLLIB")
286-
if any(custom_python["enabled"] for custom_python in self.mltask_settings["modeling"]["custom_python"]):
287-
algo_names.append("CUSTOM_PYTHON")
288-
if any(custom_python["enabled"] for custom_python in self.mltask_settings["modeling"]["plugin_python"]):
289-
algo_names.append("PLUGIN_PYTHON")
290284
return algo_names
291285

292286
def get_enabled_algorithm_settings(self):
@@ -1347,9 +1341,7 @@ def get_algorithm_settings(self, algorithm_name):
13471341
:return: A PredictionAlgorithmSettings (extended dict) for a single built-in prediction algorithm
13481342
:rtype: PredictionAlgorithmSettings
13491343
"""
1350-
if algorithm_name in ["CUSTOM_MLLIB", "CUSTOM_PYTHON", "PLUGIN_PYTHON"]:
1351-
return self.mltask_settings["modeling"][algorithm_name.lower()]
1352-
elif algorithm_name in self.__class__.algorithm_remap:
1344+
if algorithm_name in self.__class__.algorithm_remap:
13531345
algorithm_meta = self.__class__.algorithm_remap[algorithm_name]
13541346
algorithm_name = algorithm_meta.algorithm_name
13551347
algorithm_settings_class = algorithm_meta.algorithm_settings_class

0 commit comments

Comments
 (0)