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

   1 { lib
   2 , buildPythonPackage
   3 , fetchpatch
   4 , cython
   5 , numpy
   6 , pytestCheckHook
   7 , scipy
   8 , scikit-learn
   9 , fetchPypi
  10 , joblib
  11 , six
  12 , pythonRelaxDepsHook
  13 }:
  14
  15 buildPythonPackage rec {
  16   pname = "hdbscan";
  17   version = "0.8.28";
  18
  19   src = fetchPypi {
  20     inherit pname version;
  21     sha256 = "sha256-7tr3Hy87vt/Ew42hrUiXRUzl69R5LhponJecKFPtwFo=";
  22   };
  23   patches = [
  24     # should be included in next release
  25     (fetchpatch {
  26       name = "joblib-1.2.0-compat.patch";
  27       url = "https://github.com/scikit-learn-contrib/hdbscan/commit/d829c639923f6866e1917e46ddbde45b513913f3.patch";
  28       excludes = [
  29         "docs/basic_hdbscan.rst"
  30         "docs/how_hdbscan_works.rst"
  31       ];
  32       sha256 = "sha256-t0D4OsHEcMwmBZM8Mk1N0uAKi6ra+TOzEks9/efsvWI=";
  33     })
  34   ];
  35
  36   pythonRemoveDeps = [ "cython" ];
  37   nativeBuildInputs = [ pythonRelaxDepsHook cython ];
  38   propagatedBuildInputs = [ numpy scipy scikit-learn joblib six ];
  39   preCheck = ''
  40     cd hdbscan/tests
  41     rm __init__.py
  42   '';
  43   checkInputs = [ pytestCheckHook ];
  44   disabledTests = [
  45     # known flaky tests: https://github.com/scikit-learn-contrib/hdbscan/issues/420
  46     "test_mem_vec_diff_clusters"
  47     "test_all_points_mem_vec_diff_clusters"
  48     "test_approx_predict_diff_clusters"
  49     # another flaky test https://github.com/scikit-learn-contrib/hdbscan/issues/421
  50     "test_hdbscan_boruvka_balltree_matches"
  51     # more flaky tests https://github.com/scikit-learn-contrib/hdbscan/issues/570
  52     "test_hdbscan_boruvka_balltree"
  53     "test_hdbscan_best_balltree_metric"
  54   ];
  55
  56   pythonImportsCheck = [ "hdbscan" ];
  57
  58   meta = with lib; {
  59     description = "Hierarchical Density-Based Spatial Clustering of Applications with Noise, a clustering algorithm with a scikit-learn compatible API";
  60     homepage =  "https://github.com/scikit-learn-contrib/hdbscan";
  61     license = licenses.bsd3;
  62     maintainers = with maintainers; [ ixxie ];
  63   };
  64 }