export EnergyModel

Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Oct 21, 2024 · fa9a000 · fa9a000
1 parent 3e13a33
commit fa9a000
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/deepmd/jax/model/__init__.py b/deepmd/jax/model/__init__.py
@@ -1 +1,6 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
+from .ener_model import (
+    EnergyModel,
+)
+
+__all__ = ["EnergyModel"]