pkgs/development/python-modules/apricot-select/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , numba
   5 , numpy
   6 , pynose
   7 , pytestCheckHook
   8 , pythonOlder
   9 , scikit-learn
  10 , scipy
  11 , setuptools
  12 , torchvision
  13 , tqdm
  14 }:
  15
  16 buildPythonPackage rec {
  17   pname = "apricot-select";
  18   version = "0.6.1";
  19   pyproject = true;
  20
  21   disabled = pythonOlder "3.7";
  22
  23   src = fetchFromGitHub {
  24     owner = "jmschrei";
  25     repo = "apricot";
  26     rev = "refs/tags/${version}";
  27     hash = "sha256-v9BHFxmlbwXVipPze/nV35YijdFBuka3gAl85AlsffQ=";
  28   };
  29
  30   postPatch = ''
  31     sed -i '/"nose"/d' setup.py
  32   '';
  33
  34   build-system = [
  35     setuptools
  36   ];
  37
  38   dependencies = [
  39     numba
  40     numpy
  41     scipy
  42     tqdm
  43   ];
  44
  45   nativeCheckInputs = [
  46     pynose
  47     pytestCheckHook
  48     scikit-learn
  49     torchvision
  50   ];
  51
  52   pythonImportsCheck = [
  53     "apricot"
  54   ];
  55
  56   disabledTestPaths = [
  57     # Tests require nose
  58     "tests/test_optimizers/test_knapsack_facility_location.py"
  59     "tests/test_optimizers/test_knapsack_feature_based.py"
  60   ];
  61
  62   meta = with lib; {
  63     description = "Module for submodular optimization for the purpose of selecting subsets of massive data sets";
  64     homepage = "https://github.com/jmschrei/apricot";
  65     changelog = "https://github.com/jmschrei/apricot/releases/tag/${version}";
  66     license = licenses.mit;
  67     maintainers = with maintainers; [ fab ];
  68   };
  69 }