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

   1 {
   2   lib,
   3   stdenv,
   4   buildPythonPackage,
   5   fetchFromGitHub,
   6   poetry-core,
   7   poetry-dynamic-versioning,
   8   pytestCheckHook,
   9   testfixtures,
  10   anytree,
  11   beartype,
  12   gensim,
  13   graspologic-native,
  14   hyppo,
  15   joblib,
  16   matplotlib,
  17   networkx,
  18   numpy,
  19   pot,
  20   scikit-learn,
  21   scipy,
  22   seaborn,
  23   statsmodels,
  24   typing-extensions,
  25   umap-learn,
  26 }:
  27
  28 buildPythonPackage rec {
  29   pname = "graspologic";
  30   version = "3.4.1";
  31   pyproject = true;
  32
  33   src = fetchFromGitHub {
  34     owner = "graspologic-org";
  35     repo = "graspologic";
  36     rev = "refs/tags/v${version}";
  37     hash = "sha256-taX/4/uCQXW7yFykVHY78hJIGThEIycHwrEOZ3h1LPY=";
  38   };
  39
  40   build-system = [
  41     poetry-core
  42     poetry-dynamic-versioning
  43   ];
  44
  45   pythonRelaxDeps = [
  46     "beartype"
  47     "hyppo"
  48     "scipy"
  49   ];
  50
  51   dependencies = [
  52     anytree
  53     beartype
  54     gensim
  55     graspologic-native
  56     hyppo
  57     joblib
  58     matplotlib
  59     networkx
  60     numpy
  61     pot
  62     scikit-learn
  63     scipy
  64     seaborn
  65     statsmodels
  66     typing-extensions
  67     umap-learn
  68   ];
  69
  70   env.NUMBA_CACHE_DIR = "$TMPDIR";
  71
  72   nativeCheckInputs = [
  73     pytestCheckHook
  74     testfixtures
  75   ];
  76   pytestFlagsArray = [
  77     "tests"
  78     "--ignore=docs"
  79     "--ignore=tests/test_sklearn.py"
  80   ];
  81   disabledTests = [ "gridplot_outputs" ];
  82
  83   disabledTestPaths = lib.optionals stdenv.hostPlatform.isDarwin [
  84     # SIGABRT
  85     "tests/test_plot.py"
  86     "tests/test_plot_matrix.py"
  87   ];
  88
  89   meta = with lib; {
  90     description = "Package for graph statistical algorithms";
  91     homepage = "https://graspologic-org.github.io/graspologic";
  92     changelog = "https://github.com/graspologic-org/graspologic/releases/tag/v${version}";
  93     license = licenses.mit;
  94     maintainers = with maintainers; [ bcdarwin ];
  95   };
  96 }