diff --git a/setup.py b/setup.py index 7768012e9c..4be03b285b 100644 --- a/setup.py +++ b/setup.py @@ -134,9 +134,7 @@ setup( "sacremoses", ], extras_require=extras, - entry_points={ - "console_scripts": ["transformers-cli=transformers.commands.transformers_cli:main"] - }, + entry_points={"console_scripts": ["transformers-cli=transformers.commands.transformers_cli:main"]}, python_requires=">=3.6.0", classifiers=[ "Development Status :: 5 - Production/Stable", diff --git a/src/transformers/integrations.py b/src/transformers/integrations.py index 9e0ee0cbb0..27f815bd18 100644 --- a/src/transformers/integrations.py +++ b/src/transformers/integrations.py @@ -2,13 +2,11 @@ import math import os -from .file_utils import is_torch_tpu_available -from .trainer_callback import TrainerCallback -from .trainer_utils import PREFIX_CHECKPOINT_DIR, BestRun -from .utils import logging +# Import 3rd-party integrations first: try: + # Comet needs to be imported before any ML frameworks import comet_ml # noqa: F401 _has_comet = True @@ -53,6 +51,14 @@ except ImportError: except ImportError: _has_tensorboard = False +# No transformer imports above this point + +from .file_utils import is_torch_tpu_available +from .trainer_callback import TrainerCallback +from .trainer_utils import PREFIX_CHECKPOINT_DIR, BestRun +from .utils import logging + + logger = logging.get_logger(__name__)