Add Task.models property

allegroai · Mar 23, 2020 · 766c8ab · 766c8ab
1 parent 0211d23
commit 766c8ab
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/trains/task.py b/trains/task.py
@@ -458,6 +458,16 @@ def artifacts(self):
             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):
         """