split ordered by to return prediction settings for onlining

Author: kgued

dataikuapi/dss/ml.py

@@ -384,6 +384,8 @@ def split_ordered_by(self, feature_name, ascending=True):
         elif self.mltask_settings['modeling']['gridSearchParams']['mode'] == "SHUFFLE":
             self.mltask_settings['modeling']['gridSearchParams']['mode'] = "TIME_SERIES_SINGLE_SPLIT"
 
+        return self
+
     def remove_ordered_split(self):
         """
         Remove time-based ordering.
@@ -398,6 +400,8 @@ def remove_ordered_split(self):
         elif self.mltask_settings['modeling']['gridSearchParams']['mode'] == "TIME_SERIES_SINGLE_SPLIT":
             self.mltask_settings['modeling']['gridSearchParams']['mode'] = "SHUFFLE"
 
+        return self
+
     def use_sample_weighting(self, feature_name):
         """
         Uses a feature as sample weight