diff --git a/deepmd/dpmodel/__init__.py b/deepmd/dpmodel/__init__.py index 9deea3bbc0..906aac662a 100644 --- a/deepmd/dpmodel/__init__.py +++ b/deepmd/dpmodel/__init__.py @@ -14,8 +14,8 @@ OutputVariableDef, fitting_check_output, get_deriv_name, - get_reduce_name, get_hessian_name, + get_reduce_name, model_check_output, ) diff --git a/deepmd/pt/model/model/__init__.py b/deepmd/pt/model/model/__init__.py index b578419314..2bd7d0fcef 100644 --- a/deepmd/pt/model/model/__init__.py +++ b/deepmd/pt/model/model/__init__.py @@ -18,12 +18,12 @@ EnergyModel, ZBLModel, ) -from .model import ( - BaseModel, -) from .make_hessian_model import ( make_hessian_model, ) +from .model import ( + BaseModel, +) def get_zbl_model(model_params):