diff --git a/trains/backend_interface/task/task.py b/trains/backend_interface/task/task.py index a1f4cc90..cecf9539 100644 --- a/trains/backend_interface/task/task.py +++ b/trains/backend_interface/task/task.py @@ -518,7 +518,7 @@ class Task(IdObjectBase, AccessMixin, SetupUploadMixin): page=0, page_size=10, order_by='-created', - only_fields=['id'] + only_fields=['id', 'created'] ) ) model = get_single_result(entity='model', query=model_name, results=res.response.models, log=self.log) diff --git a/trains/backend_interface/util.py b/trains/backend_interface/util.py index 491815be..0a34c88b 100644 --- a/trains/backend_interface/util.py +++ b/trains/backend_interface/util.py @@ -41,9 +41,11 @@ def get_single_result(entity, query, results, log=None, show_results=10, raise_o if sort_by_date: # sort results based on timestamp and return the newest one if hasattr(results[0], 'last_update'): - results = sorted(results, key=lambda x: int(x.last_update.strftime('%s')), reverse=True) + results = sorted(results, key=lambda x: int(x.last_update.strftime('%s') + if x.last_update else 0), reverse=True) elif hasattr(results[0], 'created'): - results = sorted(results, key=lambda x: int(x.created.strftime('%s')), reverse=True) + results = sorted(results, key=lambda x: int(x.created.strftime('%s') + if x.created else 0), reverse=True) for obj in (o if isinstance(o, dict) else o.to_dict() for o in results[:show_results]): log.warn('Found {entity} `{obj[name]}` (id={obj[id]})'.format(**locals())) diff --git a/trains/model.py b/trains/model.py index cc331cf7..15cdb212 100644 --- a/trains/model.py +++ b/trains/model.py @@ -381,7 +381,7 @@ class InputModel(BaseModel): if Session.check_min_api_version('2.3') else {'tags': ["-" + ARCHIVED_TAG]} result = _Model._get_default_session().send(models.GetAllRequest( uri=[weights_url], - only_fields=["id", "name"], + only_fields=["id", "name", "created"], **extra ))