pyrox.dev
python313Packages.wandb: 0.19.8 -> 0.19.10 (#398221)
authored by
Martin Weinelt
and committed by
GitHub
e0d91bd4
ecd73d21
+19
-5
pkgs/development/python-modules/wandb/default.nix
+19
-5
pkgs/development/python-modules/wandb/default.nix
···
71
soundfile,
72
tenacity,
73
torch,
74
tqdm,
75
writableTmpDirAsHomeHook,
76
}:
77
78
let
79
-
version = "0.19.8";
80
src = fetchFromGitHub {
81
owner = "wandb";
82
repo = "wandb";
83
tag = "v${version}";
84
-
hash = "sha256-hveMyGeu9RhdtWMbV/4GQ4KUNfjSt0CKyW7Yx8QtlLM=";
85
};
86
87
gpu-stats = rustPlatform.buildRustPackage {
···
92
sourceRoot = "${src.name}/gpu_stats";
93
94
useFetchCargoVendor = true;
95
-
cargoHash = "sha256-KrwZh8OoVwImfYDmvT2Je2MYyiTZVQYngwvVC+7fTzI=";
96
97
checkFlags = [
98
# fails in sandbox
···
119
120
# hardcode the `gpu_stats` binary path.
121
postPatch = ''
122
-
substituteInPlace pkg/monitor/gpu.go \
123
--replace-fail \
124
-
'cmdPath, err := getGPUStatsCmdPath()' \
125
'cmdPath, err := "${lib.getExe gpu-stats}", error(nil)'
126
'';
127
···
136
];
137
versionCheckProgramArg = "--version";
138
doInstallCheck = true;
139
140
__darwinAllowLocalNetworking = true;
141
···
238
soundfile
239
tenacity
240
torch
241
tqdm
242
writableTmpDirAsHomeHook
243
];
···
71
soundfile,
72
tenacity,
73
torch,
74
+
torchvision,
75
tqdm,
76
writableTmpDirAsHomeHook,
77
}:
78
79
let
80
+
version = "0.19.10";
81
src = fetchFromGitHub {
82
owner = "wandb";
83
repo = "wandb";
84
tag = "v${version}";
85
+
hash = "sha256-SptjImAK0hDAr7UZjUMWnAVB1bxyzoATlyrPek5Tm48=";
86
};
87
88
gpu-stats = rustPlatform.buildRustPackage {
···
93
sourceRoot = "${src.name}/gpu_stats";
94
95
useFetchCargoVendor = true;
96
+
cargoHash = "sha256-kkWvTLduxFVIEvi4e65QQ7S0kHTRJ8XW028o430L91s=";
97
98
checkFlags = [
99
# fails in sandbox
···
120
121
# hardcode the `gpu_stats` binary path.
122
postPatch = ''
123
+
substituteInPlace pkg/monitor/gpuresourcemanager.go \
124
--replace-fail \
125
+
'cmdPath, err := getGPUCollectorCmdPath()' \
126
'cmdPath, err := "${lib.getExe gpu-stats}", error(nil)'
127
'';
128
···
137
];
138
versionCheckProgramArg = "--version";
139
doInstallCheck = true;
140
+
141
+
checkFlags =
142
+
let
143
+
skippedTests = [
144
+
# gpu sampling crashes in the sandbox
145
+
"TestSystemMonitor_BasicStateTransitions"
146
+
"TestSystemMonitor_RepeatedCalls"
147
+
"TestSystemMonitor_UnexpectedTransitions"
148
+
"TestSystemMonitor_FullCycle"
149
+
];
150
+
in
151
+
[ "-skip=^${lib.concatStringsSep "$|^" skippedTests}$" ];
152
153
__darwinAllowLocalNetworking = true;
154
···
251
soundfile
252
tenacity
253
torch
254
+
torchvision
255
tqdm
256
writableTmpDirAsHomeHook
257
];