depot/third_party/nixpkgs/pkgs/development/python-modules/mlflow/default.nix
Default email 8e65f7f0cc Project import generated by Copybara.
GitOrigin-RevId: 062a0c5437b68f950b081bbfc8a699d57a4ee026
2022-03-05 17:20:37 +01:00

77 lines
1.4 KiB
Nix

{ lib, buildPythonPackage, fetchPypi, isPy27, fetchpatch
, alembic
, click
, cloudpickle
, requests
, six
, flask
, numpy
, scipy
, pandas
, python-dateutil
, protobuf
, GitPython
, pyyaml
, querystring_parser
, simplejson
, docker
, databricks-cli
, entrypoints
, sqlparse
, sqlalchemy
, gorilla
, gunicorn
, prometheus-flask-exporter
, importlib-metadata
}:
buildPythonPackage rec {
pname = "mlflow";
version = "1.23.1";
disabled = isPy27;
src = fetchPypi {
inherit pname version;
sha256 = "03rfyhli7vbb1pz0zw75mdwj9pz3awxi3dadxn5glpwn953w6r5y";
};
# 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 buildFHSUserEnv
doCheck = false;
propagatedBuildInputs = [
alembic
click
cloudpickle
requests
six
flask
numpy
scipy
pandas
python-dateutil
protobuf
GitPython
pyyaml
querystring_parser
simplejson
docker
databricks-cli
entrypoints
sqlparse
sqlalchemy
gorilla
gunicorn
prometheus-flask-exporter
importlib-metadata
];
pythonImportsCheck = [ "mlflow" ];
meta = with lib; {
homepage = "https://github.com/mlflow/mlflow";
description = "Open source platform for the machine learning lifecycle";
license = licenses.asl20;
maintainers = with maintainers; [ tbenst ];
};
}