Skip to content

Conversation

@phoevos
Copy link
Member

@phoevos phoevos commented Dec 17, 2024

Extend the API to return MLflow tracking information as part of a training or evaluation response, including the experiment and run IDs, and update the tests accordingly. If training is already in progress, the API returns the experiment and run IDs of the current training run. This affects the following routes:

  • POST /train_supervised
  • POST /train_unsupervised
  • POST /train_unsupervised_with_hf_hub_dataset
  • POST /train_metacat
  • POST /evaluate

@phoevos phoevos added the enhancement New feature or request label Dec 17, 2024
@phoevos phoevos requested a review from baixiac December 17, 2024 19:12
@phoevos phoevos force-pushed the feature-phoevos-api-mlflow-refs branch from 44af9da to fb1d335 Compare December 18, 2024 12:42
@phoevos phoevos marked this pull request as ready for review December 18, 2024 12:43
Extend the API to return MLflow tracking information as part of a
training or evaluation response, including the experiment and run IDs,
and update the tests accordingly. If training is already in progress,
the API returns the experiment and run IDs of the current training run.
This affects the following routes:
* POST /train_supervised
* POST /train_unsupervised
* POST /train_unsupervised_with_hf_hub_dataset
* POST /train_metacat
* POST /evaluate

Signed-off-by: Phoevos Kalemkeris <phoevos.kalemkeris@ucl.ac.uk>
@phoevos phoevos force-pushed the feature-phoevos-api-mlflow-refs branch from a3bfe83 to b90b33f Compare December 18, 2024 16:42
@phoevos phoevos enabled auto-merge (squash) December 18, 2024 16:48
Copy link
Member

@baixiac baixiac left a comment

Choose a reason for hiding this comment

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

LGTM. Might be a good time to introduce a new domain class for training responses which now carry more information (instead of returning boo, str, str, return a TrainingResponse obj).

@phoevos phoevos merged commit ac0a8c1 into master Jan 6, 2025
5 checks passed
@phoevos
Copy link
Member Author

phoevos commented Jan 6, 2025

LGTM. Might be a good time to introduce a new domain class for training responses which now carry more information (instead of returning boo, str, str, return a TrainingResponse obj).

Agreed! Let's reiterate on this once we have the complete picture of how the response will be used by the upstream application (might need to add more info there).

@phoevos phoevos deleted the feature-phoevos-api-mlflow-refs branch January 6, 2025 11:03
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants