fix(pt): update import path for DP calculator in test_model_compression_se_atten

OutisLi · OutisLi · commit 3606359721e9 · 2025-11-21T15:26:42.000+08:00
diff --git a/source/tests/pt/test_model_compression_se_atten.py b/source/tests/pt/test_model_compression_se_atten.py
@@ -521,7 +521,7 @@ def test_ase(self) -> None:
             Atoms,
         )
 
-        from deepmd.tf.calculator import (
+        from deepmd.calculator import (
             DP,
         )
 

Original file line number	Diff line number	Diff line change
`@@ -521,7 +521,7 @@ def test_ase(self) -> None:`
`521`	`521`	`Atoms,`
`522`	`522`	`)`
`523`	`523`
`524`		`- from deepmd.tf.calculator import (`
	`524`	`+ from deepmd.calculator import (`
`525`	`525`	`DP,`
`526`	`526`	`)`
`527`	`527`