diff --git a/deepmd/tf/infer/deep_eval.py b/deepmd/tf/infer/deep_eval.py
index 5796dfb2a2..56df7f782f 100644
--- a/deepmd/tf/infer/deep_eval.py
+++ b/deepmd/tf/infer/deep_eval.py
@@ -1,7 +1,6 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
 import json
 from functools import (
-    cache,
     cached_property,
 )
 from typing import (
diff --git a/deepmd/utils/data_system.py b/deepmd/utils/data_system.py
index c4474aee76..7bec0b16f4 100644
--- a/deepmd/utils/data_system.py
+++ b/deepmd/utils/data_system.py
@@ -3,7 +3,6 @@
 import logging
 import warnings
 from functools import (
-    cache,
     cached_property,
 )
 from typing import (