stuebinm
ff1a94e523
The nixpkgs-unstable channel's programs.sqlite was used to identify packages producing exactly one binary, and these automatically added to their package definitions wherever possible.
113 lines
2.1 KiB
Nix
113 lines
2.1 KiB
Nix
{ lib
|
|
, alembic
|
|
, buildPythonPackage
|
|
, click
|
|
, cloudpickle
|
|
, databricks-cli
|
|
, docker
|
|
, entrypoints
|
|
, fetchPypi
|
|
, flask
|
|
, gitpython
|
|
, gorilla
|
|
, graphene
|
|
, gunicorn
|
|
, importlib-metadata
|
|
, markdown
|
|
, matplotlib
|
|
, numpy
|
|
, packaging
|
|
, pandas
|
|
, prometheus-flask-exporter
|
|
, protobuf
|
|
, python-dateutil
|
|
, pythonOlder
|
|
, pythonRelaxDepsHook
|
|
, pyarrow
|
|
, pytz
|
|
, pyyaml
|
|
, querystring-parser
|
|
, requests
|
|
, setuptools
|
|
, scikit-learn
|
|
, scipy
|
|
, simplejson
|
|
, sqlalchemy
|
|
, sqlparse
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "mlflow";
|
|
version = "2.11.1";
|
|
pyproject = true;
|
|
|
|
disabled = pythonOlder "3.8";
|
|
|
|
src = fetchPypi {
|
|
inherit pname version;
|
|
hash = "sha256-ouwp7oYvGZVSCPuOBeUeXp0u3HF143zxNtlDuYGoJOk=";
|
|
};
|
|
|
|
# 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`.
|
|
nativeBuildInputs = [
|
|
pythonRelaxDepsHook
|
|
setuptools
|
|
];
|
|
pythonRemoveDeps = [ "shap" ];
|
|
pythonRelaxDeps = [ "pytz" "pyarrow" ];
|
|
|
|
propagatedBuildInputs = [
|
|
alembic
|
|
click
|
|
cloudpickle
|
|
databricks-cli
|
|
docker
|
|
entrypoints
|
|
flask
|
|
gitpython
|
|
gorilla
|
|
graphene
|
|
gunicorn
|
|
importlib-metadata
|
|
markdown
|
|
matplotlib
|
|
numpy
|
|
packaging
|
|
pandas
|
|
prometheus-flask-exporter
|
|
protobuf
|
|
python-dateutil
|
|
pyarrow
|
|
pytz
|
|
pyyaml
|
|
querystring-parser
|
|
requests
|
|
scikit-learn
|
|
scipy
|
|
#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
|
|
doCheck = false;
|
|
|
|
meta = with lib; {
|
|
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 ];
|
|
};
|
|
}
|