Skip to content

Commit 109afbb

Browse files
committed
Fix bug in DSSMLTaskSettings new method split_ordered_by
1 parent b4fad34 commit 109afbb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dataikuapi/dss/ml.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def split_ordered_by(self, feature_name, ascending=True):
146146
:param str feature_name: Name of the variable to use
147147
:param bool ascending: True iff the test set is expected to have larger time values than the train set
148148
"""
149-
self.remove_time_variable()
149+
self.remove_ordered_split()
150150
if not feature_name in self.mltask_settings["preprocessing"]["per_feature"]:
151151
raise ValueError("Feature %s doesn't exist in this ML task, can't use as time" % feature_name)
152152
self.mltask_settings['time']['enabled'] = True

0 commit comments

Comments
 (0)