From 75d702a403426759e69c7f41a157881824a43762 Mon Sep 17 00:00:00 2001 From: Han Wang Date: Tue, 4 May 2021 10:31:53 +0800 Subject: [PATCH] fix bug: type embedding should be optional --- deepmd/utils/argcheck.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepmd/utils/argcheck.py b/deepmd/utils/argcheck.py index a4583b1b40..cc8274dea4 100644 --- a/deepmd/utils/argcheck.py +++ b/deepmd/utils/argcheck.py @@ -342,7 +342,7 @@ def model_args (): Argument("smin_alpha", float, optional = True, doc = doc_smin_alpha), Argument("sw_rmin", float, optional = True, doc = doc_sw_rmin), Argument("sw_rmax", float, optional = True, doc = doc_sw_rmax), - Argument("type_embedding", dict, type_embedding_args(), [], doc = doc_type_embedding), + Argument("type_embedding", dict, type_embedding_args(), [], optional = True, doc = doc_type_embedding), Argument("descriptor", dict, [], [descrpt_variant_type_args()], doc = doc_descrpt), Argument("fitting_net", dict, [], [fitting_variant_type_args()], doc = doc_fitting), Argument("modifier", dict, [], [modifier_variant_type_args()], optional = True, doc = doc_modifier),