iterative / mlem

🐶 A tool to package, serve, and deploy any ML model on any platform. Archived to be resurrected one day🤞
https://mlem.ai
Apache License 2.0
718 stars 44 forks source link

Bug: image_name is missing from command arguments #679

Open aShevc opened 1 year ago

aShevc commented 1 year ago

Summary / Background

Currently unable to set image_name when declaring sagemaker deployment, hence can not use prebuit images. The command attempted:


mlem declare deployment sagemaker app.mlem \
    --env env.mlem \
    --method predict \
    --use_prebuilt True \
    --instance_type ml.t2.medium \
    --env.ecr_repository 763104351884.dkr.ecr.us-east-2.amazonaws.com \
    --image_name pytorch-inference \
    --image_tag 1.12.1-cpu-py38

This returns:

No such option: --image_name (Possible options: --image_tag, --model_name)

When image_name is removed , but use_prebuilt is left intact, runnonig the command results in

image_name not specified

This blocks an ability to run prebuilt images `