Commit f51256cc authored by Weijia Wang's avatar Weijia Wang
Browse files

python311Packages.mlflow: use pyproject format

parent 6e2a59e5

pkgs/development/python-modules/mlflow/default.nix

+9 −10
Original line number Diff line number Diff line
@@ -6,11 +6,11 @@ 
, databricks-cli

, docker

, entrypoints

, fetchpatch

, fetchPypi

, flask

, gitpython

, gorilla

, graphene

, gunicorn

, importlib-metadata

, markdown
@@ -28,9 +28,9 @@ 
, pyyaml

, querystring-parser

, requests

, setuptools

, scikit-learn

, scipy

, shap

, simplejson

, sqlalchemy

, sqlparse
@@ -39,24 +39,22 @@ 
buildPythonPackage rec {

  pname = "mlflow";

  version = "2.11.1";

  format = "setuptools";

  pyproject = true;



  disabled = pythonOlder "3.7";

  disabled = pythonOlder "3.8";



  src = fetchPypi {

    inherit pname version;

    hash = "sha256-ouwp7oYvGZVSCPuOBeUeXp0u3HF143zxNtlDuYGoJOk=";

  };



  postPatch = ''

    substituteInPlace requirements/core-requirements.txt \

      --replace "gunicorn<21" "gunicorn"

  '';



  # 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 ];

  nativeBuildInputs = [

    pythonRelaxDepsHook

    setuptools

  ];

  pythonRemoveDeps = [ "shap" ];

  pythonRelaxDeps = [ "pytz" "pyarrow" ];
@@ -70,6 +68,7 @@ buildPythonPackage rec { 
    flask

    gitpython

    gorilla

    graphene

    gunicorn

    importlib-metadata

    markdown