pkgs/development/python-modules/umap-learn/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , nose
   5 , numpy
   6 , scikitlearn
   7 , scipy
   8 , numba
   9 , pynndescent
  10 , tensorflow
  11 , pytestCheckHook
  12 }:
  13
  14 buildPythonPackage rec {
  15   pname = "umap-learn";
  16   version = "0.5.0";
  17
  18   src = fetchFromGitHub {
  19     owner = "lmcinnes";
  20     repo = "umap";
  21     rev = version;
  22     sha256 = "sha256-2Z5RDi4bz8hh8zMwkcCQY9NrGaVd1DJEBOmrCl2oSvM=";
  23   };
  24
  25   checkInputs = [
  26     nose
  27     tensorflow
  28     pytestCheckHook
  29   ];
  30
  31   propagatedBuildInputs = [
  32     numpy
  33     scikitlearn
  34     scipy
  35     numba
  36     pynndescent
  37   ];
  38
  39   preCheck = ''
  40     export HOME=$TMPDIR
  41   '';
  42
  43   disabledTests = [
  44     # Plot functionality requires additional packages.
  45     # These test also fail with 'RuntimeError: cannot cache function' error.
  46     "test_umap_plot_testability"
  47     "test_plot_runs_at_all"
  48
  49     # Flaky test. Fails with AssertionError sometimes.
  50     "test_sparse_hellinger"
  51   ];
  52
  53   meta = with lib; {
  54     description = "Uniform Manifold Approximation and Projection";
  55     homepage = "https://github.com/lmcinnes/umap";
  56     license = licenses.bsd3;
  57     maintainers = [ maintainers.costrouc ];
  58   };
  59 }