Change template_script to big parameter (deepmodeling#214)

Signed-off-by: zjgemi <liuxin_zijian@163.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
zjgemi · Apr 17, 2024 · 37874a8 · 37874a8
1 parent 87f495d
commit 37874a8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/dpgen2/op/prep_dp_train.py b/dpgen2/op/prep_dp_train.py
@@ -38,7 +38,7 @@ class PrepDPTrain(OP):
  def get_input_sign(cls):
  return OPIOSign(
  {
- "template_script": Union[dict, List[dict]],
+ "template_script": BigParameter(Union[dict, List[dict]]),
  "numb_models": int,
  }
  )

diff --git a/dpgen2/superop/prep_run_dp_train.py b/dpgen2/superop/prep_run_dp_train.py
@@ -34,6 +34,7 @@
  OP,
  OPIO,
  Artifact,
+ BigParameter,
  OPIOSign,
  PythonOPTemplate,
  Slices,
@@ -76,7 +77,7 @@ def get_input_sign(cls):
  def get_output_sign(cls):
  return OPIOSign(
  {
- "template_script": List[dict],
+ "template_script": BigParameter(List[dict]),
  }
  )