Skip to content

Commit c1e8412

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent d364a27 commit c1e8412

File tree

2 files changed

+11
-5
lines changed

2 files changed

+11
-5
lines changed

dpgen/generator/arginfo.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,7 @@ def data_args() -> list[Argument]:
7878

7979
# Training
8080

81+
8182
def training_args_common() -> list[Argument]:
8283
doc_numb_models = "Number of models to be trained in 00.train. 4 is recommend."
8384
return [
@@ -240,6 +241,7 @@ def training_args() -> Variant:
240241
doc=doc_mlp_engine,
241242
)
242243

244+
243245
# Exploration
244246
def model_devi_jobs_template_args() -> Argument:
245247
doc_template = (
@@ -1004,6 +1006,10 @@ def run_jdata_arginfo() -> Argument:
10041006
"run_jdata",
10051007
dict,
10061008
sub_fields=basic_args() + data_args() + training_args_common() + fp_args(),
1007-
sub_variants=[training_args(), *model_devi_args(), fp_style_variant_type_args()],
1009+
sub_variants=[
1010+
training_args(),
1011+
*model_devi_args(),
1012+
fp_style_variant_type_args(),
1013+
],
10081014
doc=doc_run_jdata,
10091015
)

dpgen/generator/run.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ def _get_model_suffix(jdata) -> str:
140140
)
141141
return suffix
142142
else:
143-
raise ValueError("Unsupported engine: {}".format(mlp_engine))
143+
raise ValueError(f"Unsupported engine: {mlp_engine}")
144144

145145

146146
def get_job_names(jdata):
@@ -278,7 +278,7 @@ def make_train(iter_index, jdata, mdata):
278278
if mlp_engine == "dp":
279279
return make_train_dp(iter_index, jdata, mdata)
280280
else:
281-
raise ValueError("Unsupported engine: {}".format(mlp_engine))
281+
raise ValueError(f"Unsupported engine: {mlp_engine}")
282282

283283

284284
def make_train_dp(iter_index, jdata, mdata):
@@ -730,7 +730,7 @@ def run_train(iter_index, jdata, mdata):
730730
if mlp_engine == "dp":
731731
return make_train_dp(iter_index, jdata, mdata)
732732
else:
733-
raise ValueError("Unsupported engine: {}".format(mlp_engine))
733+
raise ValueError(f"Unsupported engine: {mlp_engine}")
734734

735735

736736
def run_train_dp(iter_index, jdata, mdata):
@@ -923,7 +923,7 @@ def post_train(iter_index, jdata, mdata):
923923
if mlp_engine == "dp":
924924
return post_train_dp(iter_index, jdata, mdata)
925925
else:
926-
raise ValueError("Unsupported engine: {}".format(mlp_engine))
926+
raise ValueError(f"Unsupported engine: {mlp_engine}")
927927

928928

929929
def post_train_dp(iter_index, jdata, mdata):

0 commit comments

Comments
 (0)