Skip to content

Commit 619a8c9

Browse files
authored
Remove one more deferred import (aws#46)
1 parent 36683e2 commit 619a8c9

File tree

2 files changed

+1
-4
lines changed

2 files changed

+1
-4
lines changed

src/sagemaker/session.py

-1
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@
2323
import botocore.config
2424
import six
2525
import yaml
26-
import botocore.config
2726
from botocore.exceptions import ClientError
2827

2928
from sagemaker.user_agent import prepend_user_agent

src/sagemaker/tuner.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
import inspect
1717
import json
1818

19-
from sagemaker.amazon.amazon_estimator import RecordSet
19+
from sagemaker.amazon.amazon_estimator import AmazonAlgorithmEstimatorBase, RecordSet
2020
from sagemaker.amazon.hyperparameter import Hyperparameter as hp # noqa
2121
from sagemaker.analytics import HyperparameterTuningJobAnalytics
2222
from sagemaker.estimator import Framework
@@ -114,8 +114,6 @@ def prepare_for_training(self):
114114

115115
# For attach() to know what estimator to use for non-1P algorithms
116116
# (1P algorithms don't accept extra hyperparameters)
117-
from sagemaker.amazon.amazon_estimator import AmazonAlgorithmEstimatorBase
118-
119117
if not isinstance(self.estimator, AmazonAlgorithmEstimatorBase):
120118
self.static_hyperparameters[self.SAGEMAKER_ESTIMATOR_CLASS_NAME] = json.dumps(self.estimator.__class__.__name__)
121119
self.static_hyperparameters[self.SAGEMAKER_ESTIMATOR_MODULE] = json.dumps(self.estimator.__module__)

0 commit comments

Comments
 (0)