sagemaker-xgboost-container icon indicating copy to clipboard operation
sagemaker-xgboost-container copied to clipboard

Can not load models pulled from ML Flow

Open adamwrobel-ext-gd opened this issue 2 years ago • 0 comments

Context: Models trained in SageMaker, but stored in ML Flow model registry, have multiple additional files, for example some metadata, requirements.txt, etc.

When it's zipped and provided as a SageMaker model, this function tries to read all of the files as if they were models: https://github.com/aws/sagemaker-xgboost-container/blob/b1a3cf0576fc6967584d683649d28d8897699a42/src/sagemaker_xgboost_container/algorithm_mode/handler_service.py#L52

But they are not models.

It should have some kind of pattern filter or a way to provide the file name, and not the folder. Otherwise it fails to load a model from a requirements.txt file which is just additional information.

adamwrobel-ext-gd avatar Jan 17 '23 15:01 adamwrobel-ext-gd