diff options
author | Ben Darwin <bcdarwin@gmail.com> | 2020-08-23 16:27:38 -0400 |
---|---|---|
committer | Jon <jonringer@users.noreply.github.com> | 2020-08-23 16:45:26 -0700 |
commit | 6fdfeda8b1c8d3f9a580d9cbcbc9a65b70147a02 (patch) | |
tree | 7bd631fd446834b621bea85478402dbbd8dfd4ac /pkgs/development/python-modules/test-tube | |
parent | dc20c58d90f24c0a92c649f0cf023a38b3853022 (diff) |
python3Packages.test-tube: init at 0.7.5
Diffstat (limited to 'pkgs/development/python-modules/test-tube')
-rw-r--r-- | pkgs/development/python-modules/test-tube/default.nix | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/test-tube/default.nix b/pkgs/development/python-modules/test-tube/default.nix new file mode 100644 index 0000000000000..f0c26292c84ba --- /dev/null +++ b/pkgs/development/python-modules/test-tube/default.nix @@ -0,0 +1,46 @@ +{ lib +, buildPythonPackage +, fetchFromGitHub +, isPy27 +, pytestCheckHook +, future +, imageio +, numpy +, pandas +, pytorch +, tensorflow-tensorboard +}: + +buildPythonPackage rec { + pname = "test-tube"; + version = "0.7.5"; + + disabled = isPy27; + + src = fetchFromGitHub { + owner = "williamFalcon"; + repo = pname; + rev = version; + sha256 = "0zpvlp1ybp2dhgap8jsalpfdyg8ycjhlfi3xrdf5dqffqvh2yhp2"; + }; + + checkInputs = [ + pytestCheckHook + ]; + + propagatedBuildInputs = [ + future + imageio + numpy + pandas + pytorch + tensorflow-tensorboard + ]; + + meta = with lib; { + homepage = "https://github.com/williamFalcon/test-tube"; + description = "Framework-agnostic library to track and parallelize hyperparameter search in machine learning experiments"; + license = licenses.mit; + maintainers = [ maintainers.tbenst ]; + }; +} |