nixpkgs/pkgs/development/python-modules/test-tube/default.nix
Guillaume Girol 33afbf39f6 treewide: switch to nativeCheckInputs
checkInputs used to be added to nativeBuildInputs. Now we have
nativeCheckInputs to do that instead. Doing this treewide change allows
to keep hashes identical to before the introduction of
nativeCheckInputs.
2023-01-21 12:00:00 +00:00

46 lines
833 B
Nix

{ lib
, buildPythonPackage
, fetchFromGitHub
, isPy27
, pytestCheckHook
, future
, imageio
, numpy
, pandas
, torch
, tensorboard
}:
buildPythonPackage rec {
pname = "test-tube";
version = "0.628";
disabled = isPy27;
src = fetchFromGitHub {
owner = "williamFalcon";
repo = pname;
rev = version;
sha256 = "0w60xarmcw06gc4002sy7bjfykdz34gbgniswxkl0lw8a1v0xn2m";
};
nativeCheckInputs = [
pytestCheckHook
];
propagatedBuildInputs = [
future
imageio
numpy
pandas
torch
tensorboard
];
meta = with lib; {
homepage = "https://github.com/williamFalcon/test-tube";
description = "Framework-agnostic library to track and parallelize hyperparameter search in machine learning experiments";
license = licenses.mit;
maintainers = [ maintainers.tbenst ];
};
}