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

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchPypi,
   5   fastprogress,
   6   fastcore,
   7   fastdownload,
   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.15";
  22   format = "setuptools";
  23
  24   disabled = pythonOlder "3.7";
  25
  26   src = fetchPypi {
  27     inherit pname version;
  28     hash = "sha256-8swg/RjN9exzjMVtKdVN51iH1IrRX501z1DDjTqFaSM=";
  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 = "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 }