mlflow / mlflow-export-import

Apache License 2.0
132 stars 78 forks source link

Add all entrypoints to a single Click Group #29

Open juftin opened 2 years ago

juftin commented 2 years ago

Currently this package uses a number of entrypoints to access different functionality. Moving all of these Click Commands under a Singular Click group would be much easier.

I've already performed this work on an old fork of https://github.com/amesar/mlflow-export-import/. I'll open a PR on this repo instead.

Here's what it would look like:

❯ mlflow-export-import --help
Usage: mlflow-export-import [OPTIONS] COMMAND [ARGS]...

  MLflow Export / Import CLI: Command Line Interface

Options:
  --version  Show the version and exit.
  --help     Show this message and exit.

Commands:
  export-all          Export the entire tracking server All registered...
  export-experiment   Exports an experiment to a directory.
  export-experiments  Exports experiments to a directory.
  export-model        Export a registered model and all the experiment runs...
  export-models       Exports models and their versions' backing Run along...
  export-run          Exports a run to a directory.
  find-artifacts      Find artifacts that match a filename
  http-client         Interact with the MLflow Export/Import HTTP Client
  import-experiment   Import an experiment from a directory.
  import-experiments  Import a list of experiment from a directory.
  import-model        Import a registered model and all the experiment runs...
  import-models       Imports models and their experiments and runs.
  import-run          Imports a run from a directory.
  list-models         Lists all registered models.
amesar commented 2 years ago

Sounds great. I've thought of doing this too, similar to the mlflow CLI, but time sort of ran out. Do you have a link to your fork?

juftin commented 2 years ago

Sounds great. I've thought of doing this too, similar to the mlflow CLI, but time sort of ran out. Do you have a link to your fork?

Yup, here's a PR opened from my fork: https://github.com/mlflow/mlflow-export-import/pull/30

amesar commented 2 years ago

I am on PTO until August 14, so responses might be delayed. :)