import deepmd.op in infer.data_modifier (#2592)

Fix #2590 --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
deepmodeling · Jun 6, 2023 · 03da08d · 03da08d
1 parent 6c439c7
commit 03da08d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deepmd/infer/data_modifier.py b/deepmd/infer/data_modifier.py
@@ -6,6 +6,7 @@
 
 import numpy as np
 
+import deepmd.op  # noqa: F401
 from deepmd.common import (
     make_default_mesh,
     select_idx_map,