diff options
Diffstat (limited to 'pkgs/development/python-modules/pywick/default.nix')
-rw-r--r-- | pkgs/development/python-modules/pywick/default.nix | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/pywick/default.nix b/pkgs/development/python-modules/pywick/default.nix new file mode 100644 index 000000000000..c675686ef833 --- /dev/null +++ b/pkgs/development/python-modules/pywick/default.nix @@ -0,0 +1,48 @@ +{ buildPythonPackage +, fetchFromGitHub +, pythonOlder +, pytest +, h5py +, hickle +, numpy +, pandas +, pillow +, six +, pytorch +, torchvision +, tqdm +, lib +}: + +buildPythonPackage rec { + pname = "pywick"; + version = "0.5.6"; + + disabled = pythonOlder "3.6"; + + src = fetchFromGitHub { + owner = "achaiah"; + repo = pname; + rev = "v${version}"; + sha256 = "1gmlifnv9kji0d1jwg1pa8d96zg48w17qg0sgxwy1y1jf3hn37bm"; + }; + + propagatedBuildInputs = [ + h5py hickle numpy pandas pillow six pytorch torchvision tqdm + ]; + + checkInputs = [ pytest ]; + + checkPhase = '' + runHook preCheck + pytest tests/ + runHook postCheck + ''; + + meta = { + description = "High-level training framework for Pytorch"; + homepage = "https://github.com/achaiah/pywick"; + license = lib.licenses.mit; + maintainers = with lib.maintainers; [ bcdarwin ]; + }; +} |