From 63b347db0b224a30784a946e4e78534973261109 Mon Sep 17 00:00:00 2001
From: root <2000011006@stu.pku.edu.cn>
Date: Wed, 30 Oct 2024 15:22:08 +0800
Subject: [PATCH] change tmap to all head

---
 deepmd/entrypoints/test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deepmd/entrypoints/test.py b/deepmd/entrypoints/test.py
index d9ccf392f5..fd0393c914 100644
--- 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",