autogluon / autogluon-cloud

Autogluon-cloud aims to provide user tools to train, fine-tune and deploy AutoGluon backed models on the cloud. With just a few lines of codes, users could train a model and perform inference on the cloud without worrying about MLOps details such as resource management
Apache License 2.0
18 stars 12 forks source link

Not compatible with latest sagemaker #130

Open tonyhoo opened 4 months ago

tonyhoo commented 4 months ago

With the release of SageMaker, a new field, model_reference_arn, was introduced. Refer to this PR for more details. However, it caused our integration tests to fail. The error message displayed was FAILED tests/unittests/image/test_image.py::test_multimodal_image_only - TypeError: AutoGluonNonRepackInferenceModel.prepare_container_def() got an unexpected keyword argument 'model_reference_arn'.