diff --git a/trains_agent/commands/base.py b/trains_agent/commands/base.py index d25e8ae..86482f9 100644 --- a/trains_agent/commands/base.py +++ b/trains_agent/commands/base.py @@ -12,11 +12,6 @@ from typing import Text from trains_agent.helper.console import ListFormatter, print_text from trains_agent.helper.dicts import filter_keys -try: - from typing import NoReturn -except ImportError: - from typing_extensions import NoReturn - import six from trains_agent.backend_api import services @@ -82,7 +77,7 @@ class BaseCommandSection(object): print("trains-agent: {}".format(message % args)) @classmethod - def exit(cls, message, code=1): # type: (Text, int) -> NoReturn + def exit(cls, message, code=1): # type: (Text, int) -> () cls.error(message) sys.exit(code)