diff --git a/pkgs/development/python-modules/flax/default.nix b/pkgs/development/python-modules/flax/default.nix index 50abb5eadef9..2c0f2cdaf367 100644 --- a/pkgs/development/python-modules/flax/default.nix +++ b/pkgs/development/python-modules/flax/default.nix @@ -83,6 +83,13 @@ buildPythonPackage rec { tensorflow ]; + pytestFlagsArray = [ + "-W" + # DeprecationWarning: Triggering of __jax_array__() during abstractification is deprecated. + # To avoid this error, either explicitly convert your object using jax.numpy.array(), or register your object as a pytree. + "ignore::DeprecationWarning" + ]; + disabledTestPaths = [ # Docs test, needs extra deps + we're not interested in it. "docs/_ext/codediff_test.py"