diff --git a/deepmd/tf/infer/deep_eval.py b/deepmd/tf/infer/deep_eval.py index 33725007f3..56df7f782f 100644 --- a/deepmd/tf/infer/deep_eval.py +++ b/deepmd/tf/infer/deep_eval.py @@ -1,7 +1,7 @@ # SPDX-License-Identifier: LGPL-3.0-or-later import json from functools import ( - cache, + cached_property, ) from typing import ( TYPE_CHECKING, @@ -263,8 +263,7 @@ def _init_attr(self): else: self.modifier_type = None - @property - @cache + @cached_property def model_type(self) -> type["DeepEvalWrapper"]: """Get type of model. @@ -288,8 +287,7 @@ def model_type(self) -> type["DeepEvalWrapper"]: else: raise RuntimeError(f"unknown model type {model_type}") - @property - @cache + @cached_property def model_version(self) -> str: """Get version of model. @@ -307,8 +305,7 @@ def model_version(self) -> str: [mt] = run_sess(self.sess, [t_mt], feed_dict={}) return mt.decode("utf-8") - @property - @cache + @cached_property def sess(self) -> tf.Session: """Get TF session.""" # start a tf session associated to the graph @@ -1192,8 +1189,7 @@ def __init__( self.neighbor_list = neighbor_list - @property - @cache + @cached_property def model_type(self) -> str: """Get type of model. @@ -1203,8 +1199,7 @@ def model_type(self) -> str: [mt] = run_sess(self.sess, [t_mt], feed_dict={}) return mt.decode("utf-8") - @property - @cache + @cached_property def model_version(self) -> str: """Get version of model. @@ -1222,8 +1217,7 @@ def model_version(self) -> str: [mt] = run_sess(self.sess, [t_mt], feed_dict={}) return mt.decode("utf-8") - @property - @cache + @cached_property def sess(self) -> tf.Session: """Get TF session.""" # start a tf session associated to the graph diff --git a/deepmd/tf/utils/tabulate.py b/deepmd/tf/utils/tabulate.py index afb94bb050..1dc6128f62 100644 --- a/deepmd/tf/utils/tabulate.py +++ b/deepmd/tf/utils/tabulate.py @@ -1,6 +1,7 @@ # SPDX-License-Identifier: LGPL-3.0-or-later import logging from functools import ( + cached_property, lru_cache, ) from typing import ( @@ -770,8 +771,7 @@ def _get_layer_size(self): raise RuntimeError("Unsupported descriptor") return layer_size - @property - @lru_cache + @cached_property def _n_all_excluded(self) -> int: """Then number of types excluding all types.""" return sum(int(self._all_excluded(ii)) for ii in range(0, self.ntypes)) diff --git a/deepmd/utils/data_system.py b/deepmd/utils/data_system.py index e499163e6a..7bec0b16f4 100644 --- a/deepmd/utils/data_system.py +++ b/deepmd/utils/data_system.py @@ -3,7 +3,7 @@ import logging import warnings from functools import ( - cache, + cached_property, ) from typing import ( Any, @@ -238,8 +238,7 @@ def _load_test(self, ntests=-1): for nn in test_system_data: self.test_data[nn].append(test_system_data[nn]) - @property - @cache + @cached_property def default_mesh(self) -> list[np.ndarray]: """Mesh for each system.""" return [