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

   1 { lib
   2 , buildPythonPackage
   3 , cloudpickle
   4 , dask
   5 , distributed
   6 , fetchPypi
   7 , multipledispatch
   8 , pytestCheckHook
   9 , pythonOlder
  10 , scikit-learn
  11 , scipy
  12 , setuptools-scm
  13 , sparse
  14 }:
  15
  16 buildPythonPackage rec {
  17   pname = "dask-glm";
  18   version = "0.3.2";
  19   format = "setuptools";
  20
  21   disabled = pythonOlder "3.7";
  22
  23   src = fetchPypi {
  24     inherit pname version;
  25     hash = "sha256-yUelZoZmmKAdeZeK5zIzy16DitXq1ghRQ1gsXpMLmko=";
  26   };
  27
  28   nativeBuildInputs = [
  29     setuptools-scm
  30   ];
  31
  32   propagatedBuildInputs = [
  33     cloudpickle
  34     distributed
  35     multipledispatch
  36     scikit-learn
  37     scipy
  38     sparse
  39   ] ++ dask.optional-dependencies.array;
  40
  41   nativeCheckInputs = [
  42     sparse
  43     pytestCheckHook
  44   ];
  45
  46   pythonImportsCheck = [
  47     "dask_glm"
  48   ];
  49
  50   disabledTestPaths = [
  51     # Circular dependency with dask-ml
  52     "dask_glm/tests/test_estimators.py"
  53     # Test tries to imort an obsolete method
  54     "dask_glm/tests/test_utils.py"
  55   ];
  56
  57   disabledTests = [
  58     # missing fixture with distributed>=2022.8.0
  59     "test_determinism_distributed"
  60   ];
  61
  62   meta = with lib; {
  63     description = "Generalized Linear Models with Dask";
  64     homepage = "https://github.com/dask/dask-glm/";
  65     license = licenses.bsd3;
  66     maintainers = with maintainers; [ ];
  67   };
  68 }