Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(jax): build nlist in the SavedModel & fix nopbc for StableHLO and SavedModel #4318

Merged
merged 32 commits into from
Nov 11, 2024

clean up default values

84cb819
Select commit
Loading
Failed to load commit list.
Merged

feat(jax): build nlist in the SavedModel & fix nopbc for StableHLO and SavedModel #4318

clean up default values
84cb819
Select commit
Loading
Failed to load commit list.
Codecov / codecov/patch succeeded Nov 6, 2024 in 1s

32.89% of diff hit (within 100.00% threshold of 84.42%)

View this Pull Request on Codecov

32.89% of diff hit (within 100.00% threshold of 84.42%)

Annotations

Check warning on line 78 in deepmd/jax/jax2tf/make_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/make_model.py#L73-L78 

Added lines #L73 - L78 were not covered by tests

Check warning on line 84 in deepmd/jax/jax2tf/make_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/make_model.py#L83-L84 

Added lines #L83 - L84 were not covered by tests

Check warning on line 87 in deepmd/jax/jax2tf/make_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/make_model.py#L87 

Added line #L87 was not covered by tests

Check warning on line 96 in deepmd/jax/jax2tf/make_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/make_model.py#L95-L96 

Added lines #L95 - L96 were not covered by tests

Check warning on line 104 in deepmd/jax/jax2tf/make_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/make_model.py#L104 

Added line #L104 was not covered by tests

Check warning on line 110 in deepmd/jax/jax2tf/make_model.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/make_model.py#L110 

Added line #L110 was not covered by tests

Check warning on line 61 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L59-L61 

Added lines #L59 - L61 were not covered by tests

Check warning on line 65 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L64-L65 

Added lines #L64 - L65 were not covered by tests

Check warning on line 67 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L67 

Added line #L67 was not covered by tests

Check warning on line 70 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L69-L70 

Added lines #L69 - L70 were not covered by tests

Check warning on line 78 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L73-L78 

Added lines #L73 - L78 were not covered by tests

Check warning on line 82 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L82 

Added line #L82 was not covered by tests

Check warning on line 92 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L84-L92 

Added lines #L84 - L92 were not covered by tests

Check warning on line 94 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L94 

Added line #L94 was not covered by tests

Check warning on line 98 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L98 

Added line #L98 was not covered by tests

Check warning on line 102 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L102 

Added line #L102 was not covered by tests

Check warning on line 109 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L108-L109 

Added lines #L108 - L109 were not covered by tests

Check warning on line 111 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L111 

Added line #L111 was not covered by tests

Check warning on line 135 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L123-L135 

Added lines #L123 - L135 were not covered by tests

Check warning on line 140 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L138-L140 

Added lines #L138 - L140 were not covered by tests

Check warning on line 144 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L144 

Added line #L144 was not covered by tests

Check warning on line 180 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L179-L180 

Added lines #L179 - L180 were not covered by tests

Check warning on line 188 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L182-L188 

Added lines #L182 - L188 were not covered by tests

Check warning on line 206 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L190-L206 

Added lines #L190 - L206 were not covered by tests

Check warning on line 211 in deepmd/jax/jax2tf/nlist.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

deepmd/jax/jax2tf/nlist.py#L209-L211 

Added lines #L209 - L211 were not covered by tests
View more details on Codecov