Fix

JanuszL · JanuszL · commit 510967c1b053 · 2026-03-10T10:46:46.000+01:00
Signed-off-by: Janusz Lisiecki &lt;jlisiecki@nvidia.com&gt;
diff --git a/qa/setup_packages.py b/qa/setup_packages.py
@@ -654,7 +654,7 @@ def get_pyvers_name(self, url, cuda_version):
         "flax",
         [
             PckgVer(
-                "0.7.4",
+                "0.10.0",
                 # Free-threaded Python build is incompatible with numpy<2.
                 python_free_threaded=False,
             ),

Original file line number	Diff line number	Diff line change
`@@ -654,7 +654,7 @@ def get_pyvers_name(self, url, cuda_version):`
`654`	`654`	`"flax",`
`655`	`655`	`[`
`656`	`656`	`PckgVer(`
`657`		`- "0.7.4",`
	`657`	`+ "0.10.0",`
`658`	`658`	`# Free-threaded Python build is incompatible with numpy<2.`
`659`	`659`	`python_free_threaded=False,`
`660`	`660`	`),`