fix typemap loading

theAfish · Oct 16, 2024 · f3e79dc · f3e79dc
1 parent b4d03e5
commit f3e79dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/entrypoints/test.py b/deepmd/entrypoints/test.py
@@ -127,7 +127,7 @@ def test(
         log.info(f"# testing system : {system}")
 
         # create data class
-        tmap = dp.get_type_map() if isinstance(dp, DeepPot) or isinstance(dp, DeepProperty) else None
+        tmap = dp.get_type_map() if isinstance(dp, (DeepPot, DeepProperty)) else None
         data = DeepmdData(
             system,
             set_prefix="set",