File tree Expand file tree Collapse file tree
source/tests/array_api_strict Expand file tree Collapse file tree Original file line number Diff line number Diff line change 11# SPDX-License-Identifier: LGPL-3.0-or-later
2+ from importlib import (
3+ import_module ,
4+ )
5+
26from deepmd .dpmodel .descriptor .dpa4 import DescrptDPA4 as DescrptDPA4DP
37from deepmd .dpmodel .descriptor .dpa4_nn .activation import (
48 SwiGLU ,
1923 array_api_strict_module ,
2024 register_dpmodel_mapping ,
2125)
22- from ..utils import exclude_mask as _strict_exclude_mask # noqa: F401
23- from ..utils import network as _strict_network # noqa: F401
2426from .base_descriptor import (
2527 BaseDescriptor ,
2628)
2729
30+ import_module ("..utils.exclude_mask" , __package__ )
31+ import_module ("..utils.network" , __package__ )
32+
2833
2934@BaseDescriptor .register ("SeZM" )
3035@BaseDescriptor .register ("sezm" )
Original file line number Diff line number Diff line change 11# SPDX-License-Identifier: LGPL-3.0-or-later
2+ from importlib import (
3+ import_module ,
4+ )
25from typing import (
36 ClassVar ,
47)
1316
1417from ..common import (
1518 array_api_strict_module ,
16- register_dpmodel_mapping ,
1719)
18- from ..utils import network as _strict_network # noqa: F401
20+
21+ import_module ("..utils.network" , __package__ )
1922
2023
2124@array_api_strict_module
@@ -33,14 +36,3 @@ class SeZMNetworkCollection(SeZMNetworkCollectionDP):
3336@array_api_strict_module
3437class SeZMEnergyFittingNet (SeZMEnergyFittingNetDP ):
3538 pass
36-
37-
38- register_dpmodel_mapping (
39- GLUFittingNetDP ,
40- lambda v : GLUFittingNet .deserialize (v .serialize ()),
41- )
42-
43- register_dpmodel_mapping (
44- SeZMNetworkCollectionDP ,
45- lambda v : SeZMNetworkCollection .deserialize (v .serialize ()),
46- )
You can’t perform that action at this time.
0 commit comments