pkgs/development/python-modules/test-tube/default.nix

   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
  15 buildPythonPackage 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 = pname;
  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 }