search

pmelchior / scarlet2

Scarlet, all new and shiny
MIT License
15 stars 4 forks source link

check for None in optax update #92

Closed pmelchior closed 1 week ago

pmelchior commented 1 week ago

This commit should address the problem, reported in #87, that was introduced by a change in jax 0.4.31 and another one in optax a while back.

pmelchior commented 1 week ago

Tested on python3.10 with jax 0.4.26 and optax 0.2.2 and on python3.13 with jax 0.4.34 and optax 0.2.3