tidy up imports

Author: Jonathan Esterhazy

src/sagemaker/cli/__init__.py

@@ -1,4 +1,2 @@
 import sagemaker.cli.mxnet
 import sagemaker.cli.tensorflow
-
-__all__ = [mxnet, tensorflow]

src/sagemaker/cli/common.py

@@ -97,4 +97,4 @@ def start(self):
         estimator.fit(data_url)
         logger.debug('code location: {}'.format(estimator.uploaded_code.s3_prefix))
         logger.debug('model location: {}{}/output/model.tar.gz'.format(estimator.output_path,
-                                                                      estimator._current_job_name))
+                                                                       estimator._current_job_name))

tests/unit/test_cli.py

@@ -1,7 +1,6 @@
 import pytest
-from mock import Mock, MagicMock, patch
-
 import sagemaker.cli.main as cli
+from mock import patch
 
 COMMON_ARGS = '--data mydata --script myscript --job-name myjob --bucket-name mybucket --role-name myrole ' + \
               '--python py3 --instance-type myinstance --instance-count 2'