feat(pretrained): integrate pretrained downloader into deepmd-kit#226
Closed
njzjz-bot wants to merge 3 commits intonjzjz:develfrom
Closed
feat(pretrained): integrate pretrained downloader into deepmd-kit#226njzjz-bot wants to merge 3 commits intonjzjz:develfrom
njzjz-bot wants to merge 3 commits intonjzjz:develfrom
Conversation
- add dp pretrained download <MODEL> CLI command - move pretrained logic under deepmd/pretrained - add built-in model registry with multi-source probing and fallback - register .pretrained backend alias so DeepPot usage stays unchanged - keep deep-eval adapter lazy to avoid circular imports - add parser/backend/downloader tests Authored by OpenClaw (model: custom-chat-jinzhezeng-group/gpt-5.3-codex)
| patch, | ||
| ) | ||
|
|
||
| import deepmd.backend # noqa: F401 |
Check notice
Code scanning / CodeQL
Unused import Note test
The fallback file was only added for local source-tree unittest convenience.\nKeep version behavior aligned with upstream packaging flow (_version.py via build).\n\nAuthored by OpenClaw (model: custom-chat-jinzhezeng-group/gpt-5.3-codex)
Author
|
Good catch. I have removed it now to stay aligned with upstream versioning flow. Update:
Authored by OpenClaw (model: custom-chat-jinzhezeng-group/gpt-5.3-codex) |
Authored by OpenClaw (model: custom-chat-jinzhezeng-group/gpt-5.3-codex)
Author
|
Superseded by upstream PR deepmodeling#5277. |
Codecov Report❌ Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## devel #226 +/- ##
==========================================
- Coverage 81.57% 81.48% -0.10%
==========================================
Files 750 755 +5
Lines 75216 75446 +230
Branches 3615 3616 +1
==========================================
+ Hits 61360 61475 +115
- Misses 12691 12807 +116
+ Partials 1165 1164 -1 ☔ View full report in Codecov by Sentry. 🚀 New features to boost your workflow:
|
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
This PR integrates the pretrained-dpa functionality directly into
deepmd-kitunderdeepmd/pretrained, while keepingDeepPotusage unchanged.✅ What is added
dp pretrained download <MODEL>deepmd/pretrained/registry.py,download.py,backend.py,entrypoints.pyDPA-3.2-5MDPA-3.1-3M.partwrites.pretrainedbackend alias support viadeepmd/backend/pretrained.pyDeepPot("DPA-3.2-5M.pretrained")style remains valid✅ Command parser integration
pretrainedsubcommand indeepmd/main.pydeepmd/entrypoints/main.py✅ Tests added
source/tests/common/test_pretrained_parser.pysource/tests/common/test_pretrained_download.pysource/tests/common/test_pretrained_backend.pyNotes
deepmd/__about__.pyis included to avoid import failures in source-tree runs when scm-generated version files are absent.Validation done locally
python3 -m compileallon changed modules ✅python3 -m unittest -q source.tests.common.test_pretrained_parser source.tests.common.test_pretrained_download source.tests.common.test_pretrained_backend✅ruff checkon changed files ✅Authored by OpenClaw (model: custom-chat-jinzhezeng-group/gpt-5.3-codex)