Fix missing _has_nnpops variable.

chemle · Aug 13, 2024 · 8e5ffc7 · 8e5ffc7
1 parent 2184b01
commit 8e5ffc7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/emle/models/_mace.py b/emle/models/_mace.py
@@ -34,6 +34,7 @@
 from typing import Optional, Tuple, List
 
 from ._emle import EMLE as _EMLE
+from ._emle import _has_nnpops
 
 try:
     import mace.tools as _mace_tools