pkgs/development/python-modules/dask-ml/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , dask
   4 , dask-glm
   5 , distributed
   6 , fetchPypi
   7 , multipledispatch
   8 , numba
   9 , numpy
  10 , packaging
  11 , pandas
  12 , pythonOlder
  13 , scikit-learn
  14 , scipy
  15 , setuptools-scm
  16 }:
  17
  18 buildPythonPackage rec {
  19   pname = "dask-ml";
  20   version = "2023.3.24";
  21   format = "setuptools";
  22
  23   disabled = pythonOlder "3.6";
  24
  25   src = fetchPypi {
  26     inherit pname version;
  27     hash = "sha256-lsCQ220yg2U24/Ccpk3rWZ6GRYeqjj1NLGtK9YhzMwc=";
  28   };
  29
  30   nativeBuildInputs = [
  31     setuptools-scm
  32   ];
  33
  34   propagatedBuildInputs = [
  35     dask-glm
  36     distributed
  37     multipledispatch
  38     numba
  39     numpy
  40     packaging
  41     pandas
  42     scikit-learn
  43     scipy
  44   ] ++ dask.optional-dependencies.array
  45     ++ dask.optional-dependencies.dataframe;
  46
  47   # has non-standard build from source, and pypi doesn't include tests
  48   doCheck = false;
  49
  50   pythonImportsCheck = [
  51     "dask_ml"
  52     "dask_ml.naive_bayes"
  53     "dask_ml.wrappers"
  54     "dask_ml.utils"
  55   ];
  56
  57   meta = with lib; {
  58     description = "Scalable Machine Learn with Dask";
  59     homepage = "https://github.com/dask/dask-ml";
  60     license = licenses.bsd3;
  61     maintainers = with maintainers; [ ];
  62   };
  63 }