flax

refactor: move usages of `.value` to `[...]` in modules_test.py

#4815

Merged