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

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