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

   1 { lib
   2 , fetchPypi
   3 , buildPythonPackage
   4
   5 , matplotlib
   6 , numpy
   7 , scipy
   8 , tqdm
   9 , scikit-learn
  10 , scikit-image
  11
  12 , pytestCheckHook
  13 , pythonOlder
  14 }:
  15
  16 buildPythonPackage rec {
  17   pname = "lime";
  18   version = "0.2.0.1";
  19   format = "setuptools";
  20
  21   disabled = pythonOlder "3.5";
  22
  23   src = fetchPypi {
  24     inherit pname version;
  25     hash = "sha256-dpYOTwVf61Pom1AiODuvyHtj8lusYmWYSwozPRpX94E=";
  26   };
  27
  28   propagatedBuildInputs =  [
  29     matplotlib
  30     numpy
  31     scipy
  32     tqdm
  33     scikit-learn
  34     scikit-image
  35   ];
  36
  37   nativeCheckInputs = [
  38     pytestCheckHook
  39   ];
  40
  41   disabledTestPaths = [
  42     # touches network
  43     "lime/tests/test_lime_text.py"
  44   ];
  45
  46   pythonImportsCheck = [
  47     "lime.exceptions"
  48     "lime.explanation"
  49     "lime.lime_base"
  50     "lime.lime_image"
  51     "lime.lime_text"
  52   ];
  53
  54   meta = with lib; {
  55     description = "Local Interpretable Model-Agnostic Explanations for machine learning classifiers";
  56     homepage = "https://github.com/marcotcr/lime";
  57     changelog = "https://github.com/marcotcr/lime/releases/tag/${version}";
  58     license = licenses.bsd2;
  59     maintainers = with lib.maintainers; [ khaser ];
  60   };
  61 }