Logo-amall

Hi folks! Quick question: In our MLOps infra, we are using MLFLow (version 2.0.1) as our experiment tracker. As far as I know, ZenML supports MLFLow<=1.26. Is there any plan to add support for MLFlow >= 2.0 in ZenML? Huge thanks!

Last active 15 days ago

9 replies

3 views

  • M.

    Hi folks!
    Quick question: In our MLOps infra, we are using MLFLow (version 2.0.1) as our experiment tracker. As far as I know, ZenML supports MLFLow<=1.26. Is there any plan to add support for MLFlow >= 2.0 in ZenML?
    Huge thanks!

  • HA

    hey @m.ferrer! Yes mlflow 2.0 is on the roadmap but i dont have a ETA for you yet. Can you try running the existing integration with 2.0 though? It might already work!

  • M.

    Sure @hamza, I’ll let you know the outcome!

  • JA

    Last month I ran a pipeline against mlflow 2 by mistake and it mostly worked. Things broke at the model server level (when trying to deploy the registered model), but I think the experiment tracker and model registry might work just fine

  • JA

    Here is the thread:

  • M.

    Indeed, yesterday I ran a couple of pipelines against our MLFlow deployed in the cloud (MLFlow versions 2.0.1 and 2.1.1) and both experiment tracker and model registry worked perfectly. cc @javier.lopezpena @hamza

  • M.

    Haven’t tested the model server though

  • HA

    awesome thnak you both for the input

  • HA

    We're updating mlflow to 2.0 in this PR btw

Last active 15 days ago

9 replies

3 views