about summary refs log tree commit diff
diff options
context:
space:
mode:
authorBen Darwin <bcdarwin@gmail.com>2020-01-28 16:50:39 -0500
committerJon <jonringer@users.noreply.github.com>2020-02-14 16:42:52 -0800
commit6dca90784ae865d815cb1a91d9a34661e638576b (patch)
tree91baa38ce610d2891b70e7fd7275a42ee98de9f2
parent931f47304f65e6be1c2b9ad140eb97ae7fb541b5 (diff)
downloadnixlib-6dca90784ae865d815cb1a91d9a34661e638576b.tar
nixlib-6dca90784ae865d815cb1a91d9a34661e638576b.tar.gz
nixlib-6dca90784ae865d815cb1a91d9a34661e638576b.tar.bz2
nixlib-6dca90784ae865d815cb1a91d9a34661e638576b.tar.lz
nixlib-6dca90784ae865d815cb1a91d9a34661e638576b.tar.xz
nixlib-6dca90784ae865d815cb1a91d9a34661e638576b.tar.zst
nixlib-6dca90784ae865d815cb1a91d9a34661e638576b.zip
python3Packages.pywick: init at 0.5.6
-rw-r--r--pkgs/development/python-modules/pywick/default.nix48
-rw-r--r--pkgs/top-level/python-packages.nix2
2 files changed, 50 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 ];
+  };
+}
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
index 46e5fc00b445..f13129b74d52 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -1317,6 +1317,8 @@ in {
 
   pywebpush = callPackage ../development/python-modules/pywebpush { };
 
+  pywick = callPackage ../development/python-modules/pywick { };
+
   pyxml = disabledIf isPy3k (callPackage ../development/python-modules/pyxml{ });
 
   pyvcd = callPackage ../development/python-modules/pyvcd { };