Commit c8df60c6 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files
parent 27fb23f4
Loading
Loading
Loading
Loading
+9 −4
Original line number Diff line number Diff line
@@ -4,7 +4,7 @@
  fetchFromGitHub,

  # build-system
  setuptools,
  flit-core,

  # dependencies
  absl-py,
@@ -22,18 +22,23 @@

buildPythonPackage rec {
  pname = "chex";
  version = "0.1.90";
  version = "0.1.91";
  pyproject = true;

  src = fetchFromGitHub {
    owner = "deepmind";
    repo = "chex";
    tag = "v${version}";
    hash = "sha256-nZYfkK73p3HpUOMPyqkV4YR3BDGrpdZrl36dmfZt7Bk=";
    hash = "sha256-lJ9+kvG7dRtfDVgvkcJ9/jtnX0lMfxY4mmZ290y/74U=";
  };

  build-system = [ setuptools ];
  build-system = [
    flit-core
  ];

  pythonRelaxDeps = [
    "typing_extensions"
  ];
  dependencies = [
    absl-py
    jax