diff --git a/pkgs/development/python-modules/flax/default.nix b/pkgs/development/python-modules/flax/default.nix index 7ad8ea62a125..9e9a91f0fd70 100644 --- a/pkgs/development/python-modules/flax/default.nix +++ b/pkgs/development/python-modules/flax/default.nix @@ -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 = { diff --git a/pkgs/development/python-modules/outlines/default.nix b/pkgs/development/python-modules/outlines/default.nix index eed4b70dc26d..c954547e11ec 100644 --- a/pkgs/development/python-modules/outlines/default.nix +++ b/pkgs/development/python-modules/outlines/default.nix @@ -58,6 +58,9 @@ buildPythonPackage rec { setuptools-scm ]; + pythonRelaxDeps = [ + "outlines_core" + ]; dependencies = [ cloudpickle datasets