revise based on review

dianna-ai · Jan 10, 2024 · 4ffd0cc · 4ffd0cc
1 parent 42c6ede
commit 4ffd0cc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/dianna/visualization/tabular.py b/dianna/visualization/tabular.py
@@ -21,7 +21,7 @@ def plot_tabular(
         y (List[str]): List of feature names
         x_label (str): Label for the x-axis
         y_label (str): Label or list of labels for the y-axis
-        num_features (Optional[int]): Number of top features to display
+        num_features (Optional[int]): Number of most salient features to display
         show_plot (bool, optional): Shows plot if true (for testing or writing
             plots to disk instead).
         output_filename (str, optional): Name of the file to save

diff --git a/dianna/visualization/timeseries.py b/dianna/visualization/timeseries.py
@@ -16,7 +16,7 @@ def plot_timeseries(
     x_label: str = 't',
     y_label: Union[str, Iterable[str]] = None,
     cmap: Optional[str] = None,
-    show_plot: bool = False,
+    show_plot: Optional[bool] = False,
     output_filename: Optional[str] = None,
 ) -> plt.Figure:
     """Plot timeseries with segments highlighted.