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

56 lines
970 B
Nix

{ lib
, buildPythonPackage
, fetchPypi
, fastprogress
, fastcore
, fastdownload
, torch
, torchvision
, matplotlib
, pillow
, scikit-learn
, scipy
, spacy
, pandas
, requests
, pythonOlder
}:
buildPythonPackage rec {
pname = "fastai";
version = "2.7.12";
format = "setuptools";
disabled = pythonOlder "3.7";
src = fetchPypi {
inherit pname version;
hash = "sha256-5ayA/0bdgHDGcKNI8xpkyF6hqR3DIMIQZIjzQzMoKRY=";
};
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 ];
};
}