Skip to content
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

Set up a new version of the Oumi CLI using Typer #588

Merged
merged 12 commits into from
Oct 7, 2024
Merged

Set up a new version of the Oumi CLI using Typer #588

merged 12 commits into from
Oct 7, 2024

Conversation

taenin
Copy link
Collaborator

@taenin taenin commented Oct 5, 2024

This PR creates a new oumi command that accepts 3 subcommands:
oumi train
oumi infer
oumi evaluate

These commands have full typer integration.

I will be adding oumi launch as a follow-up as it's more involved.
I will remove the old CLI when we have parity + update all references.

Towards OPE-500 and OPE-473

Copy link

linear bot commented Oct 5, 2024

OPE-500

OPE-473

@taenin taenin marked this pull request as ready for review October 5, 2024 00:02
@taenin taenin requested review from oelachqar, wizeng23 and nikg4 October 5, 2024 00:07
import typer
from typing_extensions import Annotated

import oumi.core.cli.utils as utils
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can we use a more specific name like as cli_utils ? utils is too generic

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Sure, renamed to cli_utils.

@oelachqar
Copy link
Contributor

oelachqar commented Oct 5, 2024

Would users still be able to run python -m train when the refactoring is done? or do they have to always run oumi train ?

@taenin
Copy link
Collaborator Author

taenin commented Oct 7, 2024

Would users still be able to run python -m train when the refactoring is done? or do they have to always run oumi train ?

They can still run python -m train ! This refactor only touches the CLI that we expose, not the underlying python functions.

@taenin taenin merged commit 9b62138 into main Oct 7, 2024
1 check passed
@taenin taenin deleted the taenin/cli branch October 7, 2024 17:00
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants