Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
2.3 kB · 119 lines
Nix
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120{ lib, alembic, buildPythonPackage, cachetools, click, cloudpickle, databricks-cli, docker, entrypoints, fetchPypi, flask, gitpython, gorilla, graphene, gunicorn, importlib-metadata, markdown, matplotlib, numpy, opentelemetry-api, opentelemetry-sdk, packaging, pandas, prometheus-flask-exporter, protobuf, python-dateutil, pythonOlder, pyarrow, pytz, pyyaml, querystring-parser, requests, setuptools, scikit-learn, scipy, simplejson, sqlalchemy, sqlparse,}:
buildPythonPackage rec { pname = "mlflow"; version = "2.14.2"; pyproject = true;
disabled = pythonOlder "3.8";
src = fetchPypi { inherit pname version; hash = "sha256-zqC2eK3zjR+PbNlxMKhjJddLsVk7iVtq+tx1ACHr9aI="; };
# 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 = [ setuptools ]; pythonRemoveDeps = [ "shap" ]; pythonRelaxDeps = [ "packaging" "pytz" "pyarrow" ];
propagatedBuildInputs = [ alembic cachetools click cloudpickle databricks-cli docker entrypoints flask gitpython gorilla graphene gunicorn importlib-metadata markdown matplotlib numpy opentelemetry-api opentelemetry-sdk packaging pandas prometheus-flask-exporter protobuf pyarrow python-dateutil 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 ]; };}