Skip to content

Commit 8cdfe04

Browse files
committed
addressed comments
1 parent 66cd779 commit 8cdfe04

File tree

1 file changed

+6
-11
lines changed

1 file changed

+6
-11
lines changed

src/autogluon/cloud/predictor/timeseries_cloud_predictor.py

Lines changed: 6 additions & 11 deletions
Original file line numberDiff line numberDiff 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(

0 commit comments

Comments
 (0)