Address deprecation warning in `scico.flax` #455

Merged

bwohlberg merged 2 commits intomainfrom brendt/deprec-warn

Oct 30, 2023

+5-4

Commits on Oct 5, 2023

Address DeprecationWarning: jax.random.KeyArray is deprecated. Use jax.Array for annotations
bwohlberg
committed

Commits on Oct 30, 2023

Merge branch 'main' into brendt/deprec-warn
bwohlberg
committed