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

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , isPy27
   5 , pytestCheckHook
   6 , future
   7 , imageio
   8 , numpy
   9 , pandas
  10 , pytorch
  11 , tensorflow-tensorboard
  12 }:
  13
  14 buildPythonPackage rec {
  15   pname = "test-tube";
  16   version = "0.7.5";
  17
  18   disabled = isPy27;
  19
  20   src = fetchFromGitHub {
  21     owner = "williamFalcon";
  22     repo = pname;
  23     rev = version;
  24     sha256 = "0zpvlp1ybp2dhgap8jsalpfdyg8ycjhlfi3xrdf5dqffqvh2yhp2";
  25   };
  26
  27   checkInputs = [
  28     pytestCheckHook
  29   ];
  30
  31   propagatedBuildInputs = [
  32     future
  33     imageio
  34     numpy
  35     pandas
  36     pytorch
  37     tensorflow-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 }