pkgs/development/python-modules/type-infer/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , pythonOlder
   4 , fetchPypi
   5 , poetry-core
   6 , colorlog
   7 , dataclasses-json
   8 , langid
   9 , nltk
  10 , numpy
  11 , pandas
  12 , psutil
  13 , python-dateutil
  14 , scipy
  15 , toml
  16 , nltk-data
  17 , symlinkJoin
  18 }:
  19 let
  20   testNltkData = symlinkJoin {
  21     name = "nltk-test-data";
  22     paths = [ nltk-data.punkt nltk-data.stopwords ];
  23   };
  24 in
  25 buildPythonPackage rec {
  26   pname = "type-infer";
  27   version = "0.0.18";
  28   format = "pyproject";
  29
  30   disabled = pythonOlder "3.8";
  31
  32   # using PyPI because the repo does not have tags or release branches
  33   src = fetchPypi {
  34     pname = "type_infer";
  35     inherit version;
  36     hash = "sha256-nA5TlyHpCueFWqUggS7T/eKSLlffp0pIyGCouwXPZ28=";
  37   };
  38
  39   nativeBuildInputs = [
  40     poetry-core
  41   ];
  42
  43   propagatedBuildInputs = [
  44     colorlog
  45     dataclasses-json
  46     langid
  47     nltk
  48     numpy
  49     pandas
  50     psutil
  51     python-dateutil
  52     scipy
  53     toml
  54   ];
  55
  56   # PyPI package does not include tests
  57   doCheck = false;
  58
  59   # Package import requires NLTK data to be downloaded
  60   # It is the only way to set NLTK_DATA environment variable,
  61   # so that it is available in pythonImportsCheck
  62   env.NLTK_DATA = testNltkData;
  63   pythonImportsCheck = [ "type_infer" ];
  64
  65   meta = with lib; {
  66     description = "Automated type inference for Machine Learning pipelines";
  67     homepage = "https://pypi.org/project/type-infer/";
  68     license = licenses.gpl3Only;
  69     maintainers = with maintainers; [ mbalatsko ];
  70   };
  71 }