From ab3c1ef2b912cb2afb16453831ce3678d9475714 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Thu, 19 Sep 2024 15:22:05 -0400 Subject: [PATCH] fix: move entry point from `deepmd.pt.model` to `deepmd.pt` `deepmd.pt` should be a more reasonable one to load the plugins. Signed-off-by: Jinzhe Zeng --- deepmd/pt/__init__.py | 5 +++++ deepmd/pt/model/__init__.py | 5 ----- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/deepmd/pt/__init__.py b/deepmd/pt/__init__.py index ab61736198..daf3d406e9 100644 --- a/deepmd/pt/__init__.py +++ b/deepmd/pt/__init__.py @@ -4,6 +4,11 @@ from deepmd.pt.cxx_op import ( ENABLE_CUSTOMIZED_OP, ) +from deepmd.utils.entry_point import ( + load_entry_point, +) + +load_entry_point("deepmd.pt") __all__ = [ "ENABLE_CUSTOMIZED_OP", diff --git a/deepmd/pt/model/__init__.py b/deepmd/pt/model/__init__.py index 8422ac3802..6ceb116d85 100644 --- a/deepmd/pt/model/__init__.py +++ b/deepmd/pt/model/__init__.py @@ -1,6 +1 @@ # SPDX-License-Identifier: LGPL-3.0-or-later -from deepmd.utils.entry_point import ( - load_entry_point, -) - -load_entry_point("deepmd.pt")