python312Packages.numpyro: disable failing tests (#371600)

2025-01-07 11:52:33 +01:00
parent e6ef53ea71 f73539406f
commit efff9ae7df
1 changed files with 4 additions and 0 deletions
@@ -73,6 +73,10 @@ buildPythonPackage rec {
  ];

  disabledTests = [
+    # jax.errors.UnexpectedTracerError: Encountered an unexpected tracer
+    "test_haiku_state_dropout_smoke"
+    "test_flax_state_dropout_smoke"
+
    # AssertionError due to tolerance issues
    "test_beta_binomial_log_prob"
    "test_collapse_beta"