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

   1 { lib
   2 , buildPythonPackage
   3 , fetchPypi
   4 , fastprogress
   5 , fastcore
   6 , fastdownload
   7 , torch
   8 , torchvision
   9 , matplotlib
  10 , pillow
  11 , scikit-learn
  12 , scipy
  13 , spacy
  14 , pandas
  15 , requests
  16 , pythonOlder
  17 }:
  18
  19 buildPythonPackage rec {
  20   pname = "fastai";
  21   version = "2.7.14";
  22   format = "setuptools";
  23
  24   disabled = pythonOlder "3.7";
  25
  26   src = fetchPypi {
  27     inherit pname version;
  28     hash = "sha256-QiVyYojI/xs7wYl3umhky0j+xzptNS7Fv+fLMRy/4jc=";
  29   };
  30
  31   propagatedBuildInputs = [
  32     fastprogress
  33     fastcore
  34     fastdownload
  35     torchvision
  36     matplotlib
  37     pillow
  38     scikit-learn
  39     scipy
  40     spacy
  41     pandas
  42     requests
  43   ];
  44
  45   doCheck = false;
  46   pythonImportsCheck = [ "fastai" ];
  47
  48   meta = with lib; {
  49     homepage = "https://github.com/fastai/fastai";
  50     description = "The fastai deep learning library";
  51     mainProgram = "configure_accelerate";
  52     changelog = "https://github.com/fastai/fastai/blob/${version}/CHANGELOG.md";
  53     license = licenses.asl20;
  54     maintainers = with maintainers; [ rxiao ];
  55   };
  56 }