Skip to content

Commit 3d75fcf

Browse files
Fix listing algos in py3 & remove typo
Co-authored-by: Samuel O. Ronsin <[email protected]>
1 parent dfd56aa commit 3d75fcf

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dataikuapi/dss/ml.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -386,7 +386,7 @@ def add_custom_python_model(self, name="Custom Python Model", code=""):
386386
"enabled": True
387387
})
388388

389-
def add_custom_mllib_model(self, name="Custom MLlib Model", code=""):
389+
def add_custom_mllib_model(self, name="Custom MLlib Model", code=""):
390390
"""
391391
Adds a new custom MLlib model
392392
@@ -1451,7 +1451,7 @@ def get_all_possible_algorithm_names(self):
14511451
return super(DSSPredictionMLTaskSettings, self).get_all_possible_algorithm_names() + self._get_plugin_algorithm_names()
14521452

14531453
def _get_plugin_algorithm_names(self):
1454-
return self.mltask_settings["modeling"]["plugin_python"].keys()
1454+
return list(self.mltask_settings["modeling"]["plugin_python"].keys())
14551455

14561456
def _get_plugin_algorithm_settings(self, algorithm_name):
14571457
if algorithm_name in self.mltask_settings["modeling"]["plugin_python"]:

0 commit comments

Comments
 (0)