Skip to content

Commit ffd4144

Browse files
authored
Replace persist_models with persist call due to API deprecation
1 parent a8f74e1 commit ffd4144

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/autogluon/cloud/scripts/sagemaker_scripts/tabular_serve.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def _save_image_and_update_dataframe_column(bytes):
3232
def model_fn(model_dir):
3333
"""loads model from previously saved artifact"""
3434
model = TabularPredictor.load(model_dir)
35-
model.persist_models()
35+
model.persist()
3636
globals()["column_names"] = model.original_features
3737

3838
return model

0 commit comments

Comments
 (0)