alphafold/model/layer_stack_test.py · d9e5e1d9c65ff339328c0fb7078057593eecd043 · chenzk / AlphaFold2_jax · GitLab

Find file Blame History Permalink

Fix jax.tree_multimap deprecation warning. · d9e5e1d9
Augustin Zidek authored May 31, 2022
```
PiperOrigin-RevId: 451994826
Change-Id: I4573baf61d33010c75de717d3b49f47bc9c6a8ac
```
d9e5e1d9

layer_stack_test.py 10.1 KB