diff --git a/pkgs/development/python-modules/distrax/default.nix b/pkgs/development/python-modules/distrax/default.nix index c01584c3ab4e..6c9d585a09f5 100644 --- a/pkgs/development/python-modules/distrax/default.nix +++ b/pkgs/development/python-modules/distrax/default.nix @@ -42,6 +42,9 @@ buildPythonPackage rec { # Flaky: AssertionError: 1 not less than 0.7000000000000001 "test_von_mises_sample_uniform_ks_test" + # Flaky: AssertionError: Not equal to tolerance + "test_composite_methods_are_consistent__with_jit" + # NotImplementedError: Primitive 'square' does not have a registered inverse. "test_against_tfp_bijectors_square" "test_log_dets_square__with_device"