pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 isPy27,
6 pytestCheckHook,
7 future,
8 imageio,
9 numpy,
10 pandas,
11 torch,
12 tensorboard,
13}:
14
15buildPythonPackage rec {
16 pname = "test-tube";
17 version = "0.628";
18 format = "setuptools";
19
20 disabled = isPy27;
21
22 src = fetchFromGitHub {
23 owner = "williamFalcon";
24 repo = "test-tube";
25 rev = version;
26 sha256 = "0w60xarmcw06gc4002sy7bjfykdz34gbgniswxkl0lw8a1v0xn2m";
27 };
28
29 nativeCheckInputs = [ pytestCheckHook ];
30
31 propagatedBuildInputs = [
32 future
33 imageio
34 numpy
35 pandas
36 torch
37 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}