Fix missing docs of task groups

Signed-off-by: zjgemi <liuxin_zijian@163.com>
deepmodeling · Mar 5, 2024 · 032176f · 032176f
1 parent c4d7142
commit 032176f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/docs/submit_args.rst b/docs/submit_args.rst
@@ -13,8 +13,18 @@ Arguments of the submit script
 .. _task_group_sec:
 
 Task group definition
-------------------
+---------------------
+
+LAMMPS task group
+^^^^^^^^^^^^^^^^^
+
+.. dargs::
+   :module: dpgen2.exploration.task
+   :func: lmp_task_group_args
+
+CALYPSO task group
+^^^^^^^^^^^^^^^^^^
 
 .. dargs::
    :module: dpgen2.exploration.task
-   :func: task_group_args
+   :func: caly_task_group_args
diff --git a/dpgen2/exploration/task/__init__.py b/dpgen2/exploration/task/__init__.py
@@ -17,6 +17,7 @@
     lmp_normalize as normalize_lmp_task_group_config,
 )
 from .make_task_group_from_config import (
+    caly_task_group_args,
     lmp_task_group_args,
     make_calypso_task_group_from_config,
     make_lmp_task_group_from_config,