From bb9d89b1f65b99e2e8ba8e3a6499389a5210c550 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Fri, 2 Jan 2026 21:51:58 +0000
Subject: [PATCH 1/2] python3Packages.flax: skip failing tests

---
 pkgs/development/python-modules/flax/default.nix | 12 ++++++++++++
 1 file changed, 12 insertions(+)

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 = {

From fddf575328160a58ed1e1a47ddc547a0e2237c0f Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Fri, 2 Jan 2026 22:09:04 +0000
Subject: [PATCH 2/2] python3Packages.outlines: relax outlines-core dep

---
 pkgs/development/python-modules/outlines/default.nix | 3 +++
 1 file changed, 3 insertions(+)

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