python311Packages.flax 0 -> 1
attrpath: python311Packages.flax
Checking auto update branch...
[version] 
[version] skipping because derivation has updateScript
[rustCrateVersion] 
[rustCrateVersion] No cargoSha256 or cargoHash found
[golangModuleVersion] 
[golangModuleVersion] Not a buildGoModule package with vendorSha256 or vendorHash
[npmDepsVersion] 
[npmDepsVersion] No npmDepsHash
[updateScript] 
[updateScript] Success
[updateScript] this derivation will be built:
  /nix/store/s6ak7x1pvbys7wljwi4da5j5gj9xljm2-packages.json.drv
building '/nix/store/s6ak7x1pvbys7wljwi4da5j5gj9xljm2-packages.json.drv'...

Going to be running update for following packages:
 - python3.11-flax-0.7.5

Press Enter key to continue...
Running update for:
 - python3.11-flax-0.7.5: UPDATING ...
 - python3.11-flax-0.7.5: DONE.

Packages updated!

Diff after rewrites:
diff --git a/pkgs/development/python-modules/flax/default.nix b/pkgs/development/python-modules/flax/default.nix
index b906ba9d4322..c7fc681955a3 100644
--- a/pkgs/development/python-modules/flax/default.nix
+++ b/pkgs/development/python-modules/flax/default.nix
@@ -25,7 +25,7 @@
 
 buildPythonPackage rec {
   pname = "flax";
-  version = "0.7.5";
+  version = "0.8.1";
   pyproject = true;
 
   disabled = pythonOlder "3.8";
@@ -34,7 +34,7 @@ buildPythonPackage rec {
     owner = "google";
     repo = "flax";
     rev = "refs/tags/v${version}";
-    hash = "sha256-NDah0ayQbiO1/sTU1DDf/crPq5oLTnSuosV7cFHlTM8=";
+    hash = "sha256-3UzMSJoKw+V1WLBJ+Zf7aF7CDNBsvWnRUfNgb3K4v1A=";
   };
 
   nativeBuildInputs = [

No auto update branch exists
Received ExitFailure 1 when running
Raw command: /nix/store/6r0bm8shswm9v08kzvq24ib48mx3kxmn-nix-2.18.1/bin/nix-build --option sandbox true --arg config "{ allowBroken = true; allowUnfree = true; allowAliases = false; }" --arg overlays "[ ]" -A python311Packages.flax
nix build failed.
.......................................................                  [100%]
==================================== ERRORS ====================================
_ ERROR collecting flax/experimental/nnx/examples/lm1b/input_pipeline_test.py __
ImportError while importing test module '/build/source/flax/experimental/nnx/examples/lm1b/input_pipeline_test.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
/nix/store/3v2ch16fkl50i85n05h5ckss8pxx6836-python3-3.11.8/lib/python3.11/importlib/__init__.py:126: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
flax/experimental/nnx/examples/lm1b/input_pipeline_test.py:19: in <module>
    import input_pipeline
flax/experimental/nnx/examples/lm1b/input_pipeline.py:21: in <module>
    import tensorflow_datasets as tfds
E   ModuleNotFoundError: No module named 'tensorflow_datasets'
------------------------------- Captured stderr --------------------------------
2024-03-12 16:01:19.857821: I tensorflow/core/platform/cpu_feature_guard.cc:182] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: SSE3 SSE4.1 SSE4.2 AVX AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
______ ERROR collecting flax/experimental/nnx/examples/lm1b/train_test.py ______
ImportError while importing test module '/build/source/flax/experimental/nnx/examples/lm1b/train_test.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
/nix/store/3v2ch16fkl50i85n05h5ckss8pxx6836-python3-3.11.8/lib/python3.11/importlib/__init__.py:126: in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
flax/experimental/nnx/examples/lm1b/train_test.py:20: in <module>
    import tensorflow_datasets as tfds
E   ModuleNotFoundError: No module named 'tensorflow_datasets'
=========================== short test summary info ============================
ERROR flax/experimental/nnx/examples/lm1b/input_pipeline_test.py
ERROR flax/experimental/nnx/examples/lm1b/train_test.py
================== 1493 passed, 2 skipped, 2 errors in 37.69s ==================
/nix/store/v099hqvw5z87423p4hz1vfhzaqa07dii-stdenv-linux/setup: line 1578: pop_var_context: head of shell_variables not a function context