python312Packages.mlflow: 2.14.3 -> 2.16.2

{

  lib,

  fetchFromGitHub,

  # build-system

  setuptools,

  # dependencies

  alembic,

  buildPythonPackage,

  cachetools,

  click,

  cloudpickle,

  databricks-cli,

  databricks-sdk,

  docker,

  entrypoints,

  fetchPypi,

  flask,

  gitpython,

  gorilla,

  graphene,

  gunicorn,

  importlib-metadata,

  jinja2,

  markdown,

  matplotlib,

  numpy,

  opentelemetry-sdk,

  packaging,

  pandas,

  prometheus-flask-exporter,

  protobuf,

  python-dateutil,

  pythonOlder,

  pyarrow,

  pytz,

  pyyaml,

  querystring-parser,

  requests,

  setuptools,

  scikit-learn,

  scipy,

  simplejson,

  sqlalchemy,

  sqlparse,

  # tests

  aiohttp,

  azure-core,

  azure-storage-blob,

  azure-storage-file,

  boto3,

  botocore,

  catboost,

  datasets,

  fastapi,

  google-cloud-storage,

  httpx,

  jwt,

  keras,

  langchain,

  librosa,

  moto,

  opentelemetry-exporter-otlp,

  optuna,

  pydantic,

  pyspark,

  pytestCheckHook,

  pytorch-lightning,

  sentence-transformers,

  starlette,

  statsmodels,

  tensorflow,

  torch,

  transformers,

  uvicorn,

  xgboost,

}:

buildPythonPackage rec {

  pname = "mlflow";

  version = "2.14.3";

  version = "2.16.2";

  pyproject = true;

  disabled = pythonOlder "3.8";

  src = fetchPypi {

    inherit pname version;

    hash = "sha256-KSyuS4NXSgyyIxF+IkyqZ5iTMHivAjNxnCthK+pkVhc=";

  src = fetchFromGitHub {

    owner = "mlflow";

    repo = "mlflow";

    rev = "refs/tags/v${version}";

    hash = "sha256-7W1gpVgJSN/iXoW987eCHfcOeE3D/ZJ2W/eilDdzOww=";

  };

  # 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"

    "packaging"

    "pytz"

    "pyarrow"

  build-system = [

    setuptools

  ];

  propagatedBuildInputs = [

  dependencies = [

    alembic

    cachetools

    click

    cloudpickle

    databricks-cli

    databricks-sdk

    docker

    entrypoints

    flask

    gitpython

    gorilla

    graphene

    gunicorn

    importlib-metadata

    jinja2

    markdown

    matplotlib

    numpy

    opentelemetry-sdk

    packaging

    pandas

    prometheus-flask-exporter

    protobuf

    pyarrow

    python-dateutil

    pytz

    pyyaml

    querystring-parser

    requests

    scikit-learn

    scipy

    setuptools

    #shap

    simplejson

    sqlalchemy

    sqlparse

  ];

  pythonImportsCheck = [ "mlflow" ];

  # no tests in PyPI dist

  # run into https://stackoverflow.com/questions/51203641/attributeerror-module-alembic-context-has-no-attribute-config

  # also, tests use conda so can't run on NixOS without buildFHSEnv

  nativeCheckInputs = [

    aiohttp

    azure-core

    azure-storage-blob

    azure-storage-file

    boto3

    botocore

    catboost

    datasets

    fastapi

    google-cloud-storage

    httpx

    jwt

    keras

    langchain

    librosa

    moto

    opentelemetry-exporter-otlp

    optuna

    pydantic

    pyspark

    pytestCheckHook

    pytorch-lightning

    sentence-transformers

    starlette

    statsmodels

    tensorflow

    torch

    transformers

    uvicorn

    xgboost

  ];

  disabledTestPaths = [

    # Requires unpackaged `autogen`

    "tests/autogen/test_autogen_autolog.py"

    # Requires unpackaged `diviner`

    "tests/diviner/test_diviner_model_export.py"

    # Requires unpackaged `sktime`

    "examples/sktime/test_sktime_model_export.py"

    # Requires `fastai` which would cause a circular dependency

    "tests/fastai/test_fastai_autolog.py"

    "tests/fastai/test_fastai_model_export.py"

    # Requires `spacy` which would cause a circular dependency

    "tests/spacy/test_spacy_model_export.py"

    # Requires `tensorflow.keras` which is not included in our outdated version of `tensorflow` (2.13.0)

    "tests/gateway/providers/test_ai21labs.py"

    "tests/tensorflow/test_keras_model_export.py"

    "tests/tensorflow/test_keras_pyfunc_model_works_with_all_input_types.py"

    "tests/tensorflow/test_mlflow_callback.py"

  ];

  # I (@GaetanLepage) gave up at enabling tests:

  # - They require a lot of dependencies (some unpackaged);

  # - Many errors occur at collection time;

  # - Most (all ?) tests require internet access anyway.

  doCheck = false;

  meta = with lib; {

  meta = {

    description = "Open source platform for the machine learning lifecycle";

    mainProgram = "mlflow";

    homepage = "https://github.com/mlflow/mlflow";

    changelog = "https://github.com/mlflow/mlflow/blob/v${version}/CHANGELOG.md";

    license = licenses.asl20;

    maintainers = with maintainers; [ tbenst ];

    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ tbenst ];

  };

}