feat(jax): reformat nlist in the TF model
24.13% of diff hit (within 100.00% threshold of 84.51%)
View this Pull Request on Codecov
24.13% of diff hit (within 100.00% threshold of 84.51%)
Annotations
Check warning on line 40 in deepmd/jax/jax2tf/format_nlist.py
codecov / codecov/patch
deepmd/jax/jax2tf/format_nlist.py#L38-L40
Added lines #L38 - L40 were not covered by tests
Check warning on line 42 in deepmd/jax/jax2tf/format_nlist.py
codecov / codecov/patch
deepmd/jax/jax2tf/format_nlist.py#L42
Added line #L42 was not covered by tests
Check warning on line 44 in deepmd/jax/jax2tf/format_nlist.py
codecov / codecov/patch
deepmd/jax/jax2tf/format_nlist.py#L44
Added line #L44 was not covered by tests
Check warning on line 52 in deepmd/jax/jax2tf/format_nlist.py
codecov / codecov/patch
deepmd/jax/jax2tf/format_nlist.py#L52
Added line #L52 was not covered by tests
Check warning on line 66 in deepmd/jax/jax2tf/format_nlist.py
codecov / codecov/patch
deepmd/jax/jax2tf/format_nlist.py#L54-L66
Added lines #L54 - L66 were not covered by tests
Check warning on line 68 in deepmd/jax/jax2tf/format_nlist.py
codecov / codecov/patch
deepmd/jax/jax2tf/format_nlist.py#L68
Added line #L68 was not covered by tests
Check warning on line 71 in deepmd/jax/jax2tf/format_nlist.py
codecov / codecov/patch
deepmd/jax/jax2tf/format_nlist.py#L70-L71
Added lines #L70 - L71 were not covered by tests