Commit f79856aa authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python3Packages.clu: add missing chex test dependency

parent 093d401c

pkgs/development/python-modules/clu/default.nix

+6 −4
Original line number Diff line number Diff line
@@ -19,6 +19,7 @@ 
  wrapt,



  # tests

  chex,

  keras,

  pytestCheckHook,

  tensorflow,
@@ -26,7 +27,7 @@ 
  torch,

}:



buildPythonPackage rec {

buildPythonPackage (finalAttrs: {

  pname = "clu";

  version = "0.0.12";

  pyproject = true;
@@ -34,7 +35,7 @@ buildPythonPackage rec { 
  src = fetchFromGitHub {

    owner = "google";

    repo = "CommonLoopUtils";

    tag = "v${version}";

    tag = "v${finalAttrs.version}";

    hash = "sha256-ntqRz3fCXMf0EDQsddT68Mdi105ECBVQpVsStzk2kvQ=";

  };
@@ -59,6 +60,7 @@ buildPythonPackage rec { 
  pythonImportsCheck = [ "clu" ];



  nativeCheckInputs = [

    chex

    keras

    pytestCheckHook

    tensorflow
@@ -76,8 +78,8 @@ buildPythonPackage rec { 
  meta = {

    description = "Common training loops in JAX";

    homepage = "https://github.com/google/CommonLoopUtils";

    changelog = "https://github.com/google/CommonLoopUtils/blob/${src.tag}/CHANGELOG.md";

    changelog = "https://github.com/google/CommonLoopUtils/blob/${finalAttrs.src.tag}/CHANGELOG.md";

    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ GaetanLepage ];

  };

}
 
})