change tmap to all head

Chengqian-Zhang · Chengqian-Zhang · commit 63b347db0b22 · 2024-10-30T15:22:08.000+08:00
diff --git a/deepmd/entrypoints/test.py b/deepmd/entrypoints/test.py
@@ -124,7 +124,7 @@ def test(
         log.info(f"# testing system : {system}")
 
         # create data class
-        tmap = dp.get_type_map() if isinstance(dp, DeepPot) else None
+        tmap = dp.get_type_map()
         data = DeepmdData(
             system,
             set_prefix="set",
