File tree Expand file tree Collapse file tree 1 file changed +6
-11
lines changed
src/autogluon/cloud/predictor Expand file tree Collapse file tree 1 file changed +6
-11
lines changed Original file line number Diff line number Diff line change @@ -126,20 +126,15 @@ def fit(
126126
127127 # Create predictor metadata dict
128128 predictor_metadata = {
129- "id_column" : id_column ,
130- "timestamp_column" : timestamp_column ,
131- "target_column" : predictor_init_args . get ( "target" , "target" ) ,
129+ "id_column" : self . id_column ,
130+ "timestamp_column" : self . timestamp_column ,
131+ "target_column" : self . target_column ,
132132 }
133133
134134 # Add to backend kwargs
135- if "autogluon_sagemaker_estimator_kwargs" not in backend_kwargs :
136- backend_kwargs ["autogluon_sagemaker_estimator_kwargs" ] = {}
137- if "hyperparameters" not in backend_kwargs ["autogluon_sagemaker_estimator_kwargs" ]:
138- backend_kwargs ["autogluon_sagemaker_estimator_kwargs" ]["hyperparameters" ] = {}
139-
140- backend_kwargs ["autogluon_sagemaker_estimator_kwargs" ]["hyperparameters" ]["predictor_metadata" ] = json .dumps (
141- predictor_metadata
142- )
135+ backend_kwargs .setdefault ("autogluon_sagemaker_estimator_kwargs" , {}).setdefault ("hyperparameters" , {})[
136+ "predictor_metadata"
137+ ] = json .dumps (predictor_metadata )
143138
144139 backend_kwargs = self .backend .parse_backend_fit_kwargs (backend_kwargs )
145140 self .backend .fit (
You can’t perform that action at this time.
0 commit comments