Unverified Commit 97901dc1 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python3Packages.flax: 0.10.7 -> 0.11.0 (#429462)

parents 988fe564 87553253

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

+2 −2
Original line number Diff line number Diff line
@@ -38,14 +38,14 @@ 


buildPythonPackage rec {

  pname = "flax";

  version = "0.10.7";

  version = "0.11.0";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "google";

    repo = "flax";

    tag = "v${version}";

    hash = "sha256-T/KhlvliesBW40kyToxOkyX8PLl0acMQO+FnOKqfQJo=";

    hash = "sha256-Epc7o8JoDkvNMbSH4D0cGyNJtg88qsDDbE881UVBxX4=";

  };



  build-system = [

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

+5 −0
Original line number Diff line number Diff line
@@ -92,6 +92,11 @@ buildPythonPackage rec { 


    # ValueError: compiling computation that requires 2 logical devices, but only 1 XLA devices are available (num_replicas=2)

    "test_chain"



    # Failing since flax==0.11.0

    # KeyError: "No RngStream named 'dropout' found in Rngs."

    # https://github.com/pyro-ppl/numpyro/issues/2055

    "test_nnx_state_dropout_smoke"

  ]

  ++ lib.optionals stdenv.hostPlatform.isDarwin [

    # AssertionError: Not equal to tolerance rtol=0.06, atol=0