1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
{
lib,
buildPythonPackage,
datasets,
dvc,
dvc-render,
dvc-studio-client,
fastai,
fetchFromGitHub,
funcy,
gto,
jsonargparse,
lightgbm,
lightning,
matplotlib,
mmcv,
numpy,
optuna,
pandas,
pillow,
psutil,
pynvml,
pythonOlder,
ruamel-yaml,
scikit-learn,
scmrepo,
setuptools-scm,
tensorflow,
torch,
transformers,
xgboost,
}:
buildPythonPackage rec {
pname = "dvclive";
version = "3.46.0";
pyproject = true;
disabled = pythonOlder "3.9";
src = fetchFromGitHub {
owner = "iterative";
repo = "dvclive";
rev = "refs/tags/${version}";
hash = "sha256-yIViKlkCdoG2vSZdScL38fZd9musLRKzBd9wSR6lJdk=";
};
build-system = [ setuptools-scm ];
dependencies = [
dvc
dvc-render
dvc-studio-client
funcy
gto
ruamel-yaml
scmrepo
psutil
pynvml
];
passthru.optional-dependencies = {
all = [
jsonargparse
lightgbm
lightning
matplotlib
mmcv
numpy
optuna
pandas
pillow
scikit-learn
tensorflow
torch
transformers
xgboost
] ++ jsonargparse.optional-dependencies.signatures;
image = [
numpy
pillow
];
sklearn = [ scikit-learn ];
plots = [
pandas
scikit-learn
numpy
];
markdown = [ matplotlib ];
mmcv = [ mmcv ];
tf = [ tensorflow ];
xgb = [ xgboost ];
lgbm = [ lightgbm ];
huggingface = [
datasets
transformers
];
# catalyst = [
# catalyst
# ];
fastai = [ fastai ];
lightning = [
lightning
torch
jsonargparse
] ++ jsonargparse.optional-dependencies.signatures;
optuna = [ optuna ];
};
# Circular dependency with dvc
doCheck = false;
pythonImportsCheck = [ "dvclive" ];
meta = with lib; {
description = "Library for logging machine learning metrics and other metadata in simple file formats";
homepage = "https://github.com/iterative/dvclive";
changelog = "https://github.com/iterative/dvclive/releases/tag/${version}";
license = licenses.asl20;
maintainers = with maintainers; [ fab ];
};
}
|