Skip to content

Commit 8fbb252

Browse files
committed
Fix SingleCategoryHyperparameterSettings.get_all_possible_values method
1 parent ecfa2e4 commit 8fbb252

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

dataikuapi/dss/ml.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -945,7 +945,7 @@ def get_all_possible_values(self):
945945
:return: list of possible values for this hyperparameter
946946
:rtype: list of str
947947
"""
948-
return list(self._algo_settings[self.name]["values"].keys())
948+
return self.accepted_values
949949

950950

951951
class PredictionAlgorithmSettings(dict):
@@ -1017,6 +1017,7 @@ def strategy(self, _):
10171017
"To update the search strategy, use <HyperparameterSearchSettings object>.strategy = ..., "
10181018
"obtained with <DSSPredictionMLTaskSettings object>.get_hyperparameter_search_settings()")
10191019

1020+
10201021
class RandomForestSettings(PredictionAlgorithmSettings):
10211022

10221023
def __init__(self, raw_settings, hyperparameter_search_params):

0 commit comments

Comments
 (0)