Optimize observed type retrieval in show.py

Improve efficiency by directly using observed_type from model parameters if available, avoiding unnecessary DeepEval instantiation. Update test to filter out table lines in output parsing.

Author: iProzd

deepmd/entrypoints/show.py

@@ -88,8 +88,22 @@ def show(
             total_observed_types_list = []
             model_branches = list(model_params["model_dict"].keys())
             for branch in model_branches:
-                tmp_model = DeepEval(INPUT, head=branch, no_jit=True)
-                observed_types = tmp_model.get_observed_types()
+                if (
+                    model_params["model_dict"][branch]
+                    .get("info", {})
+                    .get("observed_type", None)
+                    is not None
+                ):
+                    observed_type_list = model_params["model_dict"][branch]["info"][
+                        "observed_type"
+                    ]
+                    observed_types = {
+                        "type_num": len(observed_type_list),
+                        "observed_type": sort_element_type(observed_type_list),
+                    }
+                else:
+                    tmp_model = DeepEval(INPUT, head=branch, no_jit=True)
+                    observed_types = tmp_model.get_observed_types()
                 log.info(
                     f"{branch}: Number of observed types: {observed_types['type_num']} "
                 )

source/tests/pt/test_dp_show.py

@@ -166,8 +166,13 @@ def test_checkpoint(self) -> None:
         with redirect_stderr(io.StringIO()) as f:
             run_dp(f"dp --pt show {INPUT} {ATTRIBUTES}")
         results = [
-            res for res in f.getvalue().split("\n")[:-1] if "DEEPMD WARNING" not in res
-        ]  # filter out warnings
+            res
+            for res in f.getvalue().split("\n")[:-1]
+            if "DEEPMD WARNING" not in res
+            and "|" not in res
+            and "+-" not in res
+            and "Detailed information" not in res
+        ]  # filter out warnings and tables
         assert "This is a multitask model" in results[0]
         assert (
             "Available model branches are ['model_1', 'model_2', 'RANDOM'], "