From 7d39a7604708e36966d2f5e79074002dceef0f34 Mon Sep 17 00:00:00 2001
From: ooooo <3164076421@qq.com>
Date: Fri, 19 Jan 2024 20:19:41 +0800
Subject: [PATCH 1/2] doc107

---
 ppsci/visualize/vtu.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/ppsci/visualize/vtu.py b/ppsci/visualize/vtu.py
index 2cfde0b65..4b79e09d2 100644
--- a/ppsci/visualize/vtu.py
+++ b/ppsci/visualize/vtu.py
@@ -140,6 +140,21 @@ def save_vtu_to_mesh(
         data_dict (Dict[str, np.ndarray]): Data in dict.
         coord_keys (Tuple[str, ...]): Tuple of coord key. such as ("x", "y").
         value_keys (Tuple[str, ...]): Tuple of value key. such as ("u", "v").
+
+    Examples:
+        >>> import ppsci
+        >>> import numpy as np
+        >>> filename = "path/to/file.vtu"
+        >>> data_dict = {
+        ...     "x": np.array([[1], [2], [3],[4]]),
+        ...     "y": np.array([[2], [3], [4],[4]]),
+        ...     "z": np.array([[3], [4], [5],[4]]),
+        ...     "u": np.array([[4], [5], [6],[4]]),
+        ...     "v": np.array([[5], [6], [7],[4]]),
+        ... }
+        >>> coord_keys = ("x","y","z")
+        >>> value_keys = ("u","v")
+        >>> ppsci.visualize.save_vtu_to_mesh(filename, data_dict, coord_keys, value_keys) # doctest: +SKIP
     """
     npoint = len(next(iter(data_dict.values())))
     coord_ndim = len(coord_keys)

From 5598d4034f85aacfd2814b0a396ed5a10605fb31 Mon Sep 17 00:00:00 2001
From: ooooo <3164076421@qq.com>
Date: Tue, 23 Jan 2024 17:52:25 +0800
Subject: [PATCH 2/2] fix

---
 ppsci/visualize/vtu.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppsci/visualize/vtu.py b/ppsci/visualize/vtu.py
index 4b79e09d2..9238d4bb9 100644
--- a/ppsci/visualize/vtu.py
+++ b/ppsci/visualize/vtu.py
@@ -160,7 +160,7 @@ def save_vtu_to_mesh(
     coord_ndim = len(coord_keys)
 
     # get the list variable transposed
-    points = np.stack((data_dict[key] for key in coord_keys)).reshape(
+    points = np.stack(tuple(data_dict[key] for key in coord_keys)).reshape(
         coord_ndim, npoint
     )
     mesh = meshio.Mesh(