Be extra careful with "time" key in MLTaskSetting

Co-authored-by: Mehdi Hamoumi <[email protected]>

Author: samronsin

dataikuapi/dss/ml.py

@@ -175,7 +175,7 @@ def get_time_ordering_variable(self):
         :return: the name of the variable
         :rtype: str
         """
-        if self.mltask_settings.get('time', None) and self.mltask_settings['time']['enabled']:
+        if 'time' in self.mltask_settings and self.mltask_settings['time']['enabled']:
             return self.mltask_settings['time']['timeVariable']
         else:
             warnings.warn("Time-based ordering is disabled in the current MLTask")
@@ -185,7 +185,7 @@ def is_time_ordering_ascending(self):
         :return: True if the ordering is set to be ascending with respect to the time-ordering variable
         :rtype: bool
         """
-        if self.mltask_settings.get('time', None) and self.mltask_settings['time']['enabled']:
+        if 'time' in self.mltask_settings and self.mltask_settings['time']['enabled']:
             return self.mltask_settings['time']['ascending']
         else:
             warnings.warn("Time-based ordering is disabled in the current MLTask")