File tree 2 files changed +1
-4
lines changed
2 files changed +1
-4
lines changed Original file line number Diff line number Diff line change 23
23
import botocore .config
24
24
import six
25
25
import yaml
26
- import botocore .config
27
26
from botocore .exceptions import ClientError
28
27
29
28
from sagemaker .user_agent import prepend_user_agent
Original file line number Diff line number Diff line change 16
16
import inspect
17
17
import json
18
18
19
- from sagemaker .amazon .amazon_estimator import RecordSet
19
+ from sagemaker .amazon .amazon_estimator import AmazonAlgorithmEstimatorBase , RecordSet
20
20
from sagemaker .amazon .hyperparameter import Hyperparameter as hp # noqa
21
21
from sagemaker .analytics import HyperparameterTuningJobAnalytics
22
22
from sagemaker .estimator import Framework
@@ -114,8 +114,6 @@ def prepare_for_training(self):
114
114
115
115
# For attach() to know what estimator to use for non-1P algorithms
116
116
# (1P algorithms don't accept extra hyperparameters)
117
- from sagemaker .amazon .amazon_estimator import AmazonAlgorithmEstimatorBase
118
-
119
117
if not isinstance (self .estimator , AmazonAlgorithmEstimatorBase ):
120
118
self .static_hyperparameters [self .SAGEMAKER_ESTIMATOR_CLASS_NAME ] = json .dumps (self .estimator .__class__ .__name__ )
121
119
self .static_hyperparameters [self .SAGEMAKER_ESTIMATOR_MODULE ] = json .dumps (self .estimator .__module__ )
You can’t perform that action at this time.
0 commit comments