diff --git a/deepmd/dpmodel/atomic_model/linear_atomic_model.py b/deepmd/dpmodel/atomic_model/linear_atomic_model.py index 09b5c38263..01bfa6e7bd 100644 --- a/deepmd/dpmodel/atomic_model/linear_atomic_model.py +++ b/deepmd/dpmodel/atomic_model/linear_atomic_model.py @@ -427,7 +427,7 @@ def deserialize(cls, data) -> "DPZBLLinearEnergyAtomicModel": data.pop("@class", None) data.pop("type", None) return super().deserialize(data) - + def enable_compression( self, min_nbor_dist: float, diff --git a/deepmd/pt/model/atomic_model/linear_atomic_model.py b/deepmd/pt/model/atomic_model/linear_atomic_model.py index cc70311ea7..7ae92733d8 100644 --- a/deepmd/pt/model/atomic_model/linear_atomic_model.py +++ b/deepmd/pt/model/atomic_model/linear_atomic_model.py @@ -573,7 +573,7 @@ def deserialize(cls, data) -> "DPZBLLinearEnergyAtomicModel": data.pop("@class", None) data.pop("type", None) return super().deserialize(data) - + def enable_compression( self, min_nbor_dist: float,