diff options
Diffstat (limited to 'pkgs/development/python-modules/fastai/default.nix')
-rw-r--r-- | pkgs/development/python-modules/fastai/default.nix | 55 |
1 files changed, 55 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/fastai/default.nix b/pkgs/development/python-modules/fastai/default.nix new file mode 100644 index 0000000000000..4a4d9a16d107d --- /dev/null +++ b/pkgs/development/python-modules/fastai/default.nix @@ -0,0 +1,55 @@ +{ lib +, buildPythonPackage +, fetchPypi +, fastprogress +, fastcore +, fastdownload +, torch +, torchvision +, matplotlib +, pillow +, scikit-learn +, scipy +, spacy +, pandas +, requests +, pythonOlder +}: + +buildPythonPackage rec { + pname = "fastai"; + version = "2.7.11"; + format = "setuptools"; + + disabled = pythonOlder "3.7"; + + src = fetchPypi { + inherit pname version; + hash = "sha256-xrOhdmrCWvsPtCFDsnUxiSd7ox+Pgpmte5LyNPCHRYU="; + }; + + propagatedBuildInputs = [ + fastprogress + fastcore + fastdownload + torchvision + matplotlib + pillow + scikit-learn + scipy + spacy + pandas + requests + ]; + + doCheck = false; + pythonImportsCheck = [ "fastai" ]; + + meta = with lib; { + homepage = "https://github.com/fastai/fastai"; + description = "The fastai deep learning library"; + changelog = "https://github.com/fastai/fastai/blob/${version}/CHANGELOG.md"; + license = licenses.asl20; + maintainers = with maintainers; [ rxiao ]; + }; +} |