diff --git a/trains/__init__.py b/trains/__init__.py index 053cb431..7c711c86 100644 --- a/trains/__init__.py +++ b/trains/__init__.py @@ -4,4 +4,5 @@ from .version import __version__ from .task import Task from .model import InputModel, OutputModel, Model from .logger import Logger +from .storage import StorageManager from .errors import UsageError