Skip to content

Commit 5851b24

Browse files
committed
Rename search_mode to definition_mode, add AssertionError message and fix setter usage bug
1 parent 49c8257 commit 5851b24

1 file changed

Lines changed: 7 additions & 7 deletions

File tree

dataikuapi/dss/ml.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -583,7 +583,7 @@ def __repr__(self):
583583
def _get_active_settings_dict(self):
584584
clean_hyperparam = dict()
585585
raw_hyperparam = self._algo_settings[self.name]
586-
if self.search_mode == "EXPLICIT":
586+
if self.definition_mode == "EXPLICIT":
587587
clean_hyperparam["values"] = raw_hyperparam["values"]
588588
else:
589589
clean_hyperparam["range"] = raw_hyperparam["range"]
@@ -595,16 +595,16 @@ def _get_active_settings_dict(self):
595595
return clean_hyperparam
596596

597597
@property
598-
def search_mode(self):
598+
def definition_mode(self):
599599
if self._algo_settings.strategy == "GRID":
600600
return self._algo_settings[self.name]["gridMode"]
601601
else:
602602
# RANDOM and BAYESIAN search strategies
603603
return self._algo_settings[self.name]["randomMode"]
604604

605-
@search_mode.setter
606-
def search_mode(self, mode):
607-
assert mode in ["EXPLICIT", "RANGE"]
605+
@definition_mode.setter
606+
def definition_mode(self, mode):
607+
assert mode in ["EXPLICIT", "RANGE"], "Hyperparameter definition mode must be either \"EXPLICIT\" or \"RANGE\""
608608
if self._algo_settings.strategy == "GRID":
609609
self._algo_settings[self.name]["gridMode"] = mode
610610
else:
@@ -613,7 +613,7 @@ def search_mode(self, mode):
613613

614614
def set_explicit_values(self, values):
615615
self.values(values)
616-
self.search_mode("EXPLICIT")
616+
self.definition_mode = "EXPLICIT"
617617

618618
@property
619619
def values(self):
@@ -668,7 +668,7 @@ def _set_range(self, min=None, max=None, nb_values=None):
668668

669669
def set_range(self, min=None, max=None, nb_values=None):
670670
self._set_range(min=min, max=max, nb_values=nb_values)
671-
self.search_mode("RANGE")
671+
self.definition_mode = "RANGE"
672672

673673
@property
674674
def range(self):

0 commit comments

Comments
 (0)