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
71
soundfile,
72
72
tenacity,
73
73
torch,
74
+
torchvision,
74
75
tqdm,
75
76
writableTmpDirAsHomeHook,
76
77
}:
77
78
78
79
let
79
-
version = "0.19.8";
80
+
version = "0.19.10";
80
81
src = fetchFromGitHub {
81
82
owner = "wandb";
82
83
repo = "wandb";
83
84
tag = "v${version}";
84
-
hash = "sha256-hveMyGeu9RhdtWMbV/4GQ4KUNfjSt0CKyW7Yx8QtlLM=";
85
+
hash = "sha256-SptjImAK0hDAr7UZjUMWnAVB1bxyzoATlyrPek5Tm48=";
85
86
};
86
87
87
88
gpu-stats = rustPlatform.buildRustPackage {
···
92
93
sourceRoot = "${src.name}/gpu_stats";
93
94
94
95
useFetchCargoVendor = true;
95
-
cargoHash = "sha256-KrwZh8OoVwImfYDmvT2Je2MYyiTZVQYngwvVC+7fTzI=";
96
+
cargoHash = "sha256-kkWvTLduxFVIEvi4e65QQ7S0kHTRJ8XW028o430L91s=";
96
97
97
98
checkFlags = [
98
99
# fails in sandbox
···
119
120
120
121
# hardcode the `gpu_stats` binary path.
121
122
postPatch = ''
122
-
substituteInPlace pkg/monitor/gpu.go \
123
+
substituteInPlace pkg/monitor/gpuresourcemanager.go \
123
124
--replace-fail \
124
-
'cmdPath, err := getGPUStatsCmdPath()' \
125
+
'cmdPath, err := getGPUCollectorCmdPath()' \
125
126
'cmdPath, err := "${lib.getExe gpu-stats}", error(nil)'
126
127
'';
127
128
···
136
137
];
137
138
versionCheckProgramArg = "--version";
138
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}$" ];
139
152
140
153
__darwinAllowLocalNetworking = true;
141
154
···
238
251
soundfile
239
252
tenacity
240
253
torch
254
+
torchvision
241
255
tqdm
242
256
writableTmpDirAsHomeHook
243
257
];