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

python3Packages.flax: skip failing tests

parent bce7b104

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

+12 −0
Original line number Diff line number Diff line
@@ -107,6 +107,18 @@ buildPythonPackage rec { 


    # AssertionError: nnx_model.kernel.value.sharding = NamedSharding(...

    "test_linen_to_nnx_metadata"



    # AssertionError: 'Linear_0' not found in State({})

    "test_compact_basic"

    # KeyError: 'intermediates'

    "test_linen_submodule"

    "test_pure_nnx_submodule"

    # KeyError: 'counts

    "test_mutable_state"

    # AttributeError: 'Top' object has no attribute '_pytree__state'. Did you mean: '_pytree__flatten'?

    "test_shared_modules"

    # AttributeError: 'MLP' object has no attribute 'scope

    "test_transforms"

  ];



  passthru = {