python312Packages.mlflow: 2.19.0 -> 2.20.1

  pytestCheckHook,

  pytorch-lightning,

  sentence-transformers,

  shap,

  starlette,

  statsmodels,

  tensorflow,

buildPythonPackage rec {

  pname = "mlflow";

  version = "2.19.0";

  version = "2.20.1";

  pyproject = true;

  src = fetchFromGitHub {

    owner = "mlflow";

    repo = "mlflow";

    tag = "v${version}";

    hash = "sha256-QUyoT6tl/kv/RAntKYZ83p/lvssX1dJb45VpklAVQT4=";

    hash = "sha256-672lKzYkWpUuUB2hiVtOqMhQltzRMGhoff2ZimLWJC8=";

  };

  # Remove currently broken dependency `shap`, a model explainability package.

  # This seems quite unprincipled especially with tests not being enabled,

  # but not mlflow has a 'skinny' install option which does not require `shap`.

  pythonRemoveDeps = [ "shap" ];

  pythonRelaxDeps = [

    "gunicorn"

    "importlib-metadata"

    pandas

    protobuf

    pyarrow

    pydantic

    pyyaml

    requests

    scikit-learn

    scipy

    shap

    sqlalchemy

    sqlparse

  ];