From 766c8ab24ff7398be58dcbd9b89c762f42301efb Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Mon, 23 Mar 2020 23:25:55 +0200 Subject: [PATCH] Add Task.models property --- trains/task.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/trains/task.py b/trains/task.py index a9012487..bb7a1306 100644 --- a/trains/task.py +++ b/trains/task.py @@ -458,6 +458,16 @@ class Task(_Task): artifacts_pairs += list(self._artifacts_manager.registered_artifacts.items()) return ReadOnlyDict(artifacts_pairs) + @property + def models(self): + # type: () -> Dict[str, List[Model]] + """ + Read-only dictionary of the Task's loaded/stored models + + :return: dictionary of models loaded/stored {'input': list(Model), 'output': list(Model)} + """ + return self.get_models() + @classmethod def clone(cls, source_task=None, name=None, comment=None, parent=None, project=None): """