-
Notifications
You must be signed in to change notification settings - Fork 32
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[MRG] add CLI for convert #286
Conversation
You can refer to #288 to get model object from its ID, @STALLAAAA |
modelci/cli/modelhub.py
Outdated
|
||
@app.command('convert') | ||
def convert( | ||
id: str = typer.Option(..., '-i', '--id', help='ID of model.'), |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It's better to add more argument, such as:
- the model format use want to convert this model into, you can the use
convert
function - the saved path of the converted model
- whether to register the generated model into model hub
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I add 2 new arguments for savepath and register. I will add each specified convert after I make a revise to its function in the next PR.
72a4d7d
to
b2ce09e
Compare
:fix: add 2 arguments of convert cli :fix: make 'generate_model_family' a public function :fix:convert cli: provide args: 1. the saved path of the converted model 2. whether to register the generated model into model hub
Closes #248
What has been done to verify that this works as intended?
This is a cli command for convert component.
Why is this the best possible solution? Were any other approaches considered?
How does this change affect users? Describe intentional changes to behavior and behavior that could have accidentally been affected by code changes. In other words, what are the regression risks?
Does this change require updates to documentation?
Before submitting this PR, please make sure you have:
python -m pytest tests/
and confirmed all checks still pass.