pyrox.dev
1{ lib
2, buildPythonPackage
3, fetchFromGitHub
4, isPy27
5, pytestCheckHook
6, future
7, imageio
8, numpy
9, pandas
10, pytorch
11, tensorflow-tensorboard
12}:
13
14buildPythonPackage rec {
15 pname = "test-tube";
16 version = "0.7.5";
17
18 disabled = isPy27;
19
20 src = fetchFromGitHub {
21 owner = "williamFalcon";
22 repo = pname;
23 rev = version;
24 sha256 = "0zpvlp1ybp2dhgap8jsalpfdyg8ycjhlfi3xrdf5dqffqvh2yhp2";
25 };
26
27 checkInputs = [
28 pytestCheckHook
29 ];
30
31 propagatedBuildInputs = [
32 future
33 imageio
34 numpy
35 pandas
36 pytorch
37 tensorflow-tensorboard
38 ];
39
40 meta = with lib; {
41 homepage = "https://github.com/williamFalcon/test-tube";
42 description = "Framework-agnostic library to track and parallelize hyperparameter search in machine learning experiments";
43 license = licenses.mit;
44 maintainers = [ maintainers.tbenst ];
45 };
46}