Fix optimizer to support string

This commit is contained in:
allegroai 2020-07-30 14:51:41 +03:00
parent 29727b80b7
commit 4bfd614367

View File

@ -1087,7 +1087,7 @@ class HyperParameterOptimizer(object):
if specific_time: if specific_time:
self.optimization_timeout = specific_time.timestamp() self.optimization_timeout = specific_time.timestamp()
else: else:
self.optimization_timeout = (in_minutes * 60.) + time() if in_minutes else None self.optimization_timeout = (float(in_minutes) * 60.) + time() if in_minutes else None
def get_time_limit(self): def get_time_limit(self):
# type: () -> datetime # type: () -> datetime