Skip to content

Commit cf38369

Browse files
authored
Replace persist_models with persist call due to API deprecation (#164)
1 parent a8f74e1 commit cf38369

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
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

tests/unittests/tabular/test_tabular.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,5 +68,5 @@ def test_tabular_tabular_text_image(test_helper, framework_version):
6868
local_predictor = cloud_predictor.to_local_predictor(
6969
require_version_match=False, require_py_version_match=False
7070
)
71-
models = local_predictor.get_model_names()
71+
models = local_predictor.model_names()
7272
assert "ImagePredictor" in models

0 commit comments

Comments
 (0)