update code

Author: HydrogenSulfate

docs/zh/api/data/dataset.md

@@ -6,11 +6,14 @@
       members:
         - IterableNamedArrayDataset
         - NamedArrayDataset
-        - CylinderDataset
-        - LorenzDataset
-        - RosslerDataset
         - CSVDataset
-        - MatDataset
+        - IterableCSVDataset
         - ERA5Dataset
         - ERA5SampledDataset
+        - IterableMatDataset
+        - MatDataset
+        - CylinderDataset
+        - LorenzDataset
+        - RosslerDataset
+        - VtuDataset
       show_root_heading: false

docs/zh/api/equation.md

@@ -10,5 +10,6 @@
         - NavierStokes
         - NormalDotVec
         - Poisson
+        - Vibration
       show_root_heading: false
       heading_level: 3

docs/zh/api/geometry.md

@@ -20,25 +20,3 @@
         - TimeXGeometry
       show_root_heading: false
       heading_level: 3
-
-<!-- # Geometry
-
-::: ppsci.arch
-
-This is on a separate line
-
-$$
-\operatorname{ker} f=\{g\in G:f(g)=e_{H}\}{\mbox{.}}
-$$
-
-The homomorphism $f$ is injective if and only if its kernel is only the
-singleton set $e_G$, because otherwise $\exists a,b\in G$ with $a\neq b$ such
-that $f(a)=f(b)$.
-
-```python
-
---8<--
-./ppsci/data/dataset/array_dataset.py:16:49
---8<--
-
-``` -->

docs/zh/api/lr_scheduler.md

@@ -4,7 +4,6 @@
     handler: python
     options:
       members:
-        - ConstLR
         - Linear
         - Cosine
         - Step

docs/zh/api/metric.md

@@ -4,6 +4,7 @@
     handler: python
     options:
       members:
+        - Metric
         - MAE
         - MSE
         - RMSE

docs/zh/api/utils.md

@@ -7,7 +7,9 @@
         - initializer
         - logger
         - misc
-        - reader
+        - load_csv_file
+        - load_mat_file
+        - load_vtk_file
         - run_check
         - AttrDict
         - ExpressionSolver

ppsci/arch/base.py

@@ -62,6 +62,8 @@ def concat_to_tensor(
         Returns:
             Tuple[paddle.Tensor, ...]: Concatenated tensor.
         """
+        if len(keys) == 1:
+            return data_dict[keys[0]]
         data = [data_dict[key] for key in keys]
         return paddle.concat(data, axis)
 
@@ -78,6 +80,8 @@ def split_to_dict(
         Returns:
             Dict[str, paddle.Tensor]: Dict contains tensor.
         """
+        if len(keys) == 1:
+            return {keys[0]: data_tensor}
         data = paddle.split(data_tensor, len(keys), axis=axis)
         return {key: data[i] for i, key in enumerate(keys)}
 

ppsci/utils/__init__.py

@@ -22,6 +22,10 @@
 from ppsci.utils.expression import ExpressionSolver
 from ppsci.utils.misc import AverageMeter
 from ppsci.utils.misc import set_random_seed
+from ppsci.utils.reader import load_csv_file
+from ppsci.utils.reader import load_mat_file
+from ppsci.utils.reader import load_vtk_file
+from ppsci.utils.reader import load_vtk_with_time_file
 from ppsci.utils.save_load import load_checkpoint
 from ppsci.utils.save_load import load_pretrain
 from ppsci.utils.save_load import save_checkpoint
@@ -31,6 +35,10 @@
     "logger",
     "misc",
     "reader",
+    "load_csv_file",
+    "load_mat_file",
+    "load_vtk_file",
+    "load_vtk_with_time_file",
     "dynamic_import_to_globals",
     "run_check",
     "AttrDict",

ppsci/utils/reader.py

@@ -27,6 +27,13 @@
 
 from ppsci.utils import logger
 
+__all__ = [
+    "load_csv_file",
+    "load_mat_file",
+    "load_vtk_file",
+    "load_vtk_with_time_file",
+]
+
 
 def load_csv_file(
     file_path: str,
@@ -119,7 +126,7 @@ def load_vtk_file(
     input_keys: Tuple[str, ...],
     label_keys: Optional[Tuple[str, ...]],
 ) -> Dict[str, np.ndarray]:
-    """load coordinates and attached label from the vtu file
+    """load coordinates and attached label from the *.vtu file.
 
     Args:
         filename_without_timeid (str): File name without time id.

ppsci/visualize/visualizer.py

@@ -291,7 +291,6 @@ class Visualizer3D(base.Visualizer):
         batch_size (int, optional): Batch size of data when computing result in visu.py. Defaults to 64.
         label_dict (Dict[str, np.ndarray]): Label dict.
         time_list (Optional[Tuple[float, ...]]): Time list.
-        num_timestamps (int, optional): Number of timestamps.
         prefix (str, optional): Prefix for output file.
     """