Skip to content

Commit

Permalink
add deepmd-kit multi-task parameters (#50)
Browse files Browse the repository at this point in the history
  • Loading branch information
njzjz authored Dec 12, 2024
1 parent 254f19a commit dfdb1d1
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,10 @@
import dpgen
import dpgen2
import dpdispatcher
from deepmd.utils.argcheck import gen_args as deepmd_arginfo
from deepmd.utils.argcheck import (
gen_args as deepmd_arginfo,
gen_args_multi_task as deepmd_multi_task_arginfo,
)
from dpgen.generator.arginfo import run_jdata_arginfo, run_mdata_arginfo
from dpgen.simplify.arginfo import simplify_jdata_arginfo, simplify_mdata_arginfo
from dpgen.data.arginfo import (
Expand Down Expand Up @@ -96,6 +99,13 @@ def add_parameter(key : str, name: str, argument: Argument, version: str,
}
],
)
add_parameter(
"deepmd-kit-multi-task",
"DeePMD-kit",
deepmd_multi_task_arginfo(),
deepmd.__version__,
"deepmodeling/deepmd-kit",
)

# dpgen
add_parameter(
Expand Down

0 comments on commit dfdb1d1

Please sign in to comment.