Description
Is your feature request related to a problem? Please describe.
Metadata extracted from Merlin is missing some information such as the model_signature
, params
and metrics
used to train the model. This information is present in MLFlow and the MLFlow endpoint is available in the Merlin response.
Describe the solution you'd like
Integrate MLFlow in addition to Merlin and enhance the extracted asset with additional information present in MLFlow.
Additional context
Although there is a different issue (#450) for adding an extractor for MLFlow, the MLFlow extractor cannot be directly used to fulfil our objective(because the URN generated by it would be different that the one generated by Merlin). We might be able to reuse the MLFLow client so it might be a good idea to address this issue after #450.