From 2126408dbe4ebf8c56c7a8319dc5be70f983c1bb Mon Sep 17 00:00:00 2001 From: Shaochen Shi Date: Thu, 5 Aug 2021 10:58:12 +0800 Subject: [PATCH] Fix member declartion of `deepmd` and `deepmd.entrypoints`. --- deepmd/__init__.py | 2 +- deepmd/entrypoints/__init__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/deepmd/__init__.py b/deepmd/__init__.py index 64386b754c..3a295dcbef 100644 --- a/deepmd/__init__.py +++ b/deepmd/__init__.py @@ -4,7 +4,7 @@ from . import cluster, descriptor, fit, loss, utils from .env import set_mkl -from .infer import DeepPotential +from .infer import DeepEval, DeepPotential from .infer.data_modifier import DipoleChargeModifier set_mkl() diff --git a/deepmd/entrypoints/__init__.py b/deepmd/entrypoints/__init__.py index d92cbf3076..80d0431f8c 100644 --- a/deepmd/entrypoints/__init__.py +++ b/deepmd/entrypoints/__init__.py @@ -17,7 +17,7 @@ "doc_train_input", "freeze", "test", - "train", + "train_dp", "transfer", "compress", "doc_train_input",