Commit e87f642c authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files
parent 9b69e322
Loading
Loading
Loading
Loading
+4 −7
Original line number Diff line number Diff line
@@ -10,7 +10,6 @@

  ## gpu-stats
  rustPlatform,
  darwin,

  ## wandb
  buildPythonPackage,
@@ -72,15 +71,16 @@
  tenacity,
  torch,
  tqdm,
  writableTmpDirAsHomeHook,
}:

let
  version = "0.19.5";
  version = "0.19.6";
  src = fetchFromGitHub {
    owner = "wandb";
    repo = "wandb";
    tag = "v${version}";
    hash = "sha256-NkwXHogHBsDdaWdATWujdvnSr0oXp3RbWq5hQX53WR8=";
    hash = "sha256-snyr0IlE4otk1ctWUrJEFAmHYsXe+k6qULCaO3aW0e4=";
  };

  gpu-stats = rustPlatform.buildRustPackage {
@@ -236,12 +236,9 @@ buildPythonPackage rec {
    tenacity
    torch
    tqdm
    writableTmpDirAsHomeHook
  ];

  preCheck = ''
    export HOME=$(mktemp -d)
  '';

  # test_matplotlib_image_with_multiple_axes may take >60s
  pytestFlagsArray = [
    "--timeout=1024"