scitex 2.3.0__py3-none-any.whl → 2.4.1__py3-none-any.whl

scitex/plt/_subplots/_export_as_csv_formatters/_format_plot.py

@@ -1,44 +1,66 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
-# Timestamp: "2025-05-18 18:14:26 (ywatanabe)"
-# File: /data/gpfs/projects/punim2354/ywatanabe/scitex_repo/src/scitex/plt/_subplots/_export_as_csv_formatters/_format_plot.py
-# ----------------------------------------
-import os
-__FILE__ = __file__
-__DIR__ = os.path.dirname(__FILE__)
-# ----------------------------------------
+# Timestamp: "2025-12-01 13:30:00 (ywatanabe)"
+# File: ./src/scitex/plt/_subplots/_export_as_csv_formatters/_format_plot.py
+
+"""CSV formatter for matplotlib plot() calls."""
 
 from collections import OrderedDict
+from typing import Any, Dict, Optional
+
 import numpy as np
 import pandas as pd
 import xarray as xr
 
-def _format_plot(id, tracked_dict, kwargs):
-    """Format data from a plot call.
 
-    Args:
-        id (str): Identifier for the plot
-        tracked_dict (dict): Dictionary containing tracked data
-        kwargs (dict): Keyword arguments passed to plot
+def _format_plot(
+    id: str,
+    tracked_dict: Optional[Dict[str, Any]],
+    kwargs: Dict[str, Any],
+) -> pd.DataFrame:
+    """Format data from a plot() call for CSV export.
+
+    Handles various input formats including:
+    - Pre-formatted plot_df from scitex wrappers
+    - Raw args from __getattr__ proxied matplotlib calls
+    - Single array: plot(y) generates x from indices
+    - Two arrays: plot(x, y)
+    - 2D arrays: creates multiple x/y column pairs
 
-    Returns:
-        pd.DataFrame: Formatted data from plot
+    Parameters
+    ----------
+    id : str
+        Identifier prefix for the output columns (e.g., "ax_00").
+    tracked_dict : dict or None
+        Dictionary containing tracked data. May include:
+        - 'plot_df': Pre-formatted DataFrame from wrapper
+        - 'args': Raw positional arguments (x, y) from plot()
+    kwargs : dict
+        Keyword arguments passed to plot (currently unused).
+
+    Returns
+    -------
+    pd.DataFrame
+        Formatted data with columns prefixed by id.
+        For 1D data: {id}_plot_x, {id}_plot_y
+        For 2D data: {id}_plot_x00, {id}_plot_y00, {id}_plot_x01, ...
     """
     # Check if tracked_dict is empty or not a dictionary
     if not tracked_dict or not isinstance(tracked_dict, dict):
         return pd.DataFrame()
-    
-    # For plot_line, we expect a 'plot_df' key
+
+    # For stx_line, we expect a 'plot_df' key
     if 'plot_df' in tracked_dict:
         plot_df = tracked_dict['plot_df']
         if isinstance(plot_df, pd.DataFrame):
             # Add the id prefix to all columns
             return plot_df.add_prefix(f"{id}_")
-    
-    # Legacy handling for tracked args (should be deprecated)
+
+    # Handle raw args from __getattr__ proxied calls
     if 'args' in tracked_dict:
         args = tracked_dict['args']
         if isinstance(args, tuple) and len(args) > 0:
+            # Handle single argument: plot(y) or plot(data_2d)
             if len(args) == 1:
                 args_value = args[0]
 
@@ -60,32 +82,47 @@ def _format_plot(id, tracked_dict, kwargs):
                     df = pd.DataFrame({f"{id}_plot_x": x, f"{id}_plot_y": y})
                     return df
 
-            elif len(args) == 2:
-                x, y = args
-                if isinstance(y, (np.ndarray, xr.DataArray)):
-                    if y.ndim == 2:
-                        out = OrderedDict()
-                        for ii in range(y.shape[1]):
-                            out[f"{id}_plot_x{ii:02d}"] = x
-                            out[f"{id}_plot_y{ii:02d}"] = y[:, ii]
-                        df = pd.DataFrame(out)
-                        return df
-
-                if isinstance(y, pd.DataFrame):
+            # Handle two arguments: plot(x, y)
+            elif len(args) >= 2:
+                x_arg, y_arg = args[0], args[1]
+
+                # Convert to numpy
+                x = np.asarray(x_arg.values if hasattr(x_arg, 'values') else x_arg)
+                y = np.asarray(y_arg.values if hasattr(y_arg, 'values') else y_arg)
+
+                # Handle 2D y array (multiple lines)
+                if hasattr(y, 'ndim') and y.ndim == 2:
+                    out = OrderedDict()
+                    for ii in range(y.shape[1]):
+                        out[f"{id}_plot_x{ii:02d}"] = x
+                        out[f"{id}_plot_y{ii:02d}"] = y[:, ii]
+                    df = pd.DataFrame(out)
+                    return df
+
+                # Handle DataFrame y
+                if isinstance(y_arg, pd.DataFrame):
                     df = pd.DataFrame(
                         {
                             f"{id}_plot_x": x,
                             **{
-                                f"{id}_plot_y{ii:02d}": np.array(y[col])
-                                for ii, col in enumerate(y.columns)
+                                f"{id}_plot_y{ii:02d}": np.array(y_arg[col])
+                                for ii, col in enumerate(y_arg.columns)
                             },
                         }
                     )
                     return df
 
-                if isinstance(y, (np.ndarray, xr.DataArray, list)):
-                    df = pd.DataFrame({f"{id}_plot_x": x, f"{id}_plot_y": y})
+                # Handle 1D arrays (most common case: plot(x, y))
+                if hasattr(y, 'ndim') and y.ndim == 1:
+                    # Flatten x if needed
+                    x_flat = np.ravel(x)
+                    y_flat = np.ravel(y)
+                    df = pd.DataFrame({f"{id}_plot_x": x_flat, f"{id}_plot_y": y_flat})
                     return df
 
+                # Fallback for list-like y
+                df = pd.DataFrame({f"{id}_plot_x": np.ravel(x), f"{id}_plot_y": np.ravel(y)})
+                return df
+
     # Default empty DataFrame if we can't process the input
     return pd.DataFrame()

scitex/plt/_subplots/_export_as_csv_formatters/_format_plot_box.py

@@ -12,7 +12,7 @@ import numpy as np
 import pandas as pd
 
 def _format_plot_box(id, tracked_dict, kwargs):
-    """Format data from a plot_box call."""
+    """Format data from a stx_box call."""
     # Check if tracked_dict is empty or not a dictionary
     if not tracked_dict or not isinstance(tracked_dict, dict):
         return pd.DataFrame()

scitex/plt/_subplots/_export_as_csv_formatters/_format_plot_kde.py

@@ -12,14 +12,14 @@ import pandas as pd
 from scitex.pd import force_df
 
 def _format_plot_kde(id, tracked_dict, kwargs):
-    """Format data from a plot_kde call.
+    """Format data from a stx_kde call.
     
     Processes kernel density estimation plot data.
     
     Args:
         id (str): Identifier for the plot
         tracked_dict (dict): Dictionary containing 'x', 'kde', and 'n' keys
-        kwargs (dict): Keyword arguments passed to plot_kde
+        kwargs (dict): Keyword arguments passed to stx_kde
         
     Returns:
         pd.DataFrame: Formatted KDE data

scitex/plt/_subplots/_export_as_csv_formatters/_format_quiver.py

@@ -0,0 +1,53 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Timestamp: "2025-12-01 12:20:00 (ywatanabe)"
+# File: /home/ywatanabe/proj/scitex-code/src/scitex/plt/_subplots/_export_as_csv_formatters/_format_quiver.py
+
+import numpy as np
+import pandas as pd
+
+
+def _format_quiver(id, tracked_dict, kwargs):
+    """Format data from a quiver (vector field) call.
+
+    Args:
+        id (str): Identifier for the plot
+        tracked_dict (dict): Dictionary containing tracked data
+        kwargs (dict): Keyword arguments passed to quiver
+
+    Returns:
+        pd.DataFrame: Formatted data from quiver (X, Y positions and U, V vectors)
+    """
+    if not tracked_dict or not isinstance(tracked_dict, dict):
+        return pd.DataFrame()
+
+    if 'args' in tracked_dict:
+        args = tracked_dict['args']
+        if isinstance(args, tuple):
+            # quiver can be called as:
+            # quiver(U, V) - positions auto-generated
+            # quiver(X, Y, U, V) - explicit positions
+            if len(args) == 2:
+                U = np.asarray(args[0])
+                V = np.asarray(args[1])
+                X, Y = np.meshgrid(np.arange(U.shape[1]), np.arange(U.shape[0]))
+            elif len(args) >= 4:
+                X = np.asarray(args[0])
+                Y = np.asarray(args[1])
+                U = np.asarray(args[2])
+                V = np.asarray(args[3])
+            else:
+                return pd.DataFrame()
+
+            df = pd.DataFrame({
+                f"{id}_quiver_x": X.flatten(),
+                f"{id}_quiver_y": Y.flatten(),
+                f"{id}_quiver_u": U.flatten(),
+                f"{id}_quiver_v": V.flatten()
+            })
+            return df
+
+    return pd.DataFrame()
+
+
+# EOF

scitex/plt/_subplots/_export_as_csv_formatters/_format_stem.py

@@ -0,0 +1,42 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Timestamp: "2025-12-01 12:20:00 (ywatanabe)"
+# File: /home/ywatanabe/proj/scitex-code/src/scitex/plt/_subplots/_export_as_csv_formatters/_format_stem.py
+
+import numpy as np
+import pandas as pd
+
+
+def _format_stem(id, tracked_dict, kwargs):
+    """Format data from a stem plot call.
+
+    Args:
+        id (str): Identifier for the plot
+        tracked_dict (dict): Dictionary containing tracked data
+        kwargs (dict): Keyword arguments passed to stem
+
+    Returns:
+        pd.DataFrame: Formatted data from stem plot
+    """
+    if not tracked_dict or not isinstance(tracked_dict, dict):
+        return pd.DataFrame()
+
+    if 'args' in tracked_dict:
+        args = tracked_dict['args']
+        if isinstance(args, tuple) and len(args) > 0:
+            if len(args) == 1:
+                y = np.asarray(args[0])
+                x = np.arange(len(y))
+            elif len(args) >= 2:
+                x = np.asarray(args[0])
+                y = np.asarray(args[1])
+            else:
+                return pd.DataFrame()
+
+            df = pd.DataFrame({f"{id}_stem_x": x, f"{id}_stem_y": y})
+            return df
+
+    return pd.DataFrame()
+
+
+# EOF

scitex/plt/_subplots/_export_as_csv_formatters/_format_step.py

@@ -0,0 +1,42 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Timestamp: "2025-12-01 12:20:00 (ywatanabe)"
+# File: /home/ywatanabe/proj/scitex-code/src/scitex/plt/_subplots/_export_as_csv_formatters/_format_step.py
+
+import numpy as np
+import pandas as pd
+
+
+def _format_step(id, tracked_dict, kwargs):
+    """Format data from a step plot call.
+
+    Args:
+        id (str): Identifier for the plot
+        tracked_dict (dict): Dictionary containing tracked data
+        kwargs (dict): Keyword arguments passed to step
+
+    Returns:
+        pd.DataFrame: Formatted data from step plot
+    """
+    if not tracked_dict or not isinstance(tracked_dict, dict):
+        return pd.DataFrame()
+
+    if 'args' in tracked_dict:
+        args = tracked_dict['args']
+        if isinstance(args, tuple) and len(args) > 0:
+            if len(args) == 1:
+                y = np.asarray(args[0])
+                x = np.arange(len(y))
+            elif len(args) >= 2:
+                x = np.asarray(args[0])
+                y = np.asarray(args[1])
+            else:
+                return pd.DataFrame()
+
+            df = pd.DataFrame({f"{id}_step_x": x, f"{id}_step_y": y})
+            return df
+
+    return pd.DataFrame()
+
+
+# EOF

scitex/plt/_subplots/_export_as_csv_formatters/_format_streamplot.py

@@ -0,0 +1,48 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# Timestamp: "2025-12-01 12:20:00 (ywatanabe)"
+# File: /home/ywatanabe/proj/scitex-code/src/scitex/plt/_subplots/_export_as_csv_formatters/_format_streamplot.py
+
+import numpy as np
+import pandas as pd
+
+
+def _format_streamplot(id, tracked_dict, kwargs):
+    """Format data from a streamplot call.
+
+    Args:
+        id (str): Identifier for the plot
+        tracked_dict (dict): Dictionary containing tracked data
+        kwargs (dict): Keyword arguments passed to streamplot
+
+    Returns:
+        pd.DataFrame: Formatted data from streamplot (X, Y positions and U, V vectors)
+    """
+    if not tracked_dict or not isinstance(tracked_dict, dict):
+        return pd.DataFrame()
+
+    if 'args' in tracked_dict:
+        args = tracked_dict['args']
+        if isinstance(args, tuple) and len(args) >= 4:
+            # streamplot(X, Y, U, V) - X, Y are 1D, U, V are 2D
+            X = np.asarray(args[0])
+            Y = np.asarray(args[1])
+            U = np.asarray(args[2])
+            V = np.asarray(args[3])
+
+            # Create meshgrid if X, Y are 1D
+            if X.ndim == 1 and Y.ndim == 1:
+                X, Y = np.meshgrid(X, Y)
+
+            df = pd.DataFrame({
+                f"{id}_streamplot_x": X.flatten(),
+                f"{id}_streamplot_y": Y.flatten(),
+                f"{id}_streamplot_u": U.flatten(),
+                f"{id}_streamplot_v": V.flatten()
+            })
+            return df
+
+    return pd.DataFrame()
+
+
+# EOF

scitex/plt/_subplots/_export_as_csv_formatters/{_format_plot_conf_mat.py → _format_stx_conf_mat.py}

@@ -12,12 +12,12 @@ import numpy as np
 import pandas as pd
 
 def _format_plot_conf_mat(id, tracked_dict, kwargs):
-    """Format data from a plot_conf_mat call.
+    """Format data from a stx_conf_mat call.
     
     Args:
         id (str): Identifier for the plot
         tracked_dict (dict): Dictionary containing tracked data
-        kwargs (dict): Keyword arguments passed to plot_conf_mat
+        kwargs (dict): Keyword arguments passed to stx_conf_mat
         
     Returns:
         pd.DataFrame: Formatted confusion matrix data

scitex/plt/_subplots/_export_as_csv_formatters/{_format_plot_ecdf.py → _format_stx_ecdf.py}

@@ -11,12 +11,12 @@ __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 
 def _format_plot_ecdf(id, tracked_dict, kwargs):
-    """Format data from a plot_ecdf call.
+    """Format data from a stx_ecdf call.
     
     Args:
         id (str): Identifier for the plot
         tracked_dict (dict): Dictionary containing 'ecdf_df' key with ECDF data
-        kwargs (dict): Keyword arguments passed to plot_ecdf
+        kwargs (dict): Keyword arguments passed to stx_ecdf
         
     Returns:
         pd.DataFrame: Formatted ECDF data

scitex/plt/_subplots/_export_as_csv_formatters/{_format_plot_fillv.py → _format_stx_fillv.py}

@@ -12,14 +12,14 @@ import numpy as np
 import pandas as pd
 
 def _format_plot_fillv(id, tracked_dict, kwargs):
-    """Format data from a plot_fillv call.
+    """Format data from a stx_fillv call.
     
     Formats data similar to line plot format for better compatibility.
     
     Args:
         id (str): Identifier for the plot
         tracked_dict (dict): Dictionary containing tracked data
-        kwargs (dict): Keyword arguments passed to plot_fillv
+        kwargs (dict): Keyword arguments passed to stx_fillv
         
     Returns:
         pd.DataFrame: Formatted fillv data in a long-format dataframe

scitex/plt/_subplots/_export_as_csv_formatters/{_format_plot_heatmap.py → _format_stx_heatmap.py}

@@ -12,14 +12,14 @@ import numpy as np
 import pandas as pd
 
 def _format_plot_heatmap(id, tracked_dict, kwargs):
-    """Format data from a plot_heatmap call.
+    """Format data from a stx_heatmap call.
     
     Exports heatmap data in xyz format (x, y, value) for better compatibility.
     
     Args:
         id (str): Identifier for the plot
         tracked_dict (dict): Dictionary containing tracked data
-        kwargs (dict): Keyword arguments passed to plot_heatmap
+        kwargs (dict): Keyword arguments passed to stx_heatmap
         
     Returns:
         pd.DataFrame: Formatted heatmap data in xyz format

scitex/plt/_subplots/_export_as_csv_formatters/{_format_plot_image.py → _format_stx_image.py}

@@ -12,7 +12,7 @@ __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 
 def _format_plot_image(id, tracked_dict, kwargs):
-    """Format data from a plot_image call.
+    """Format data from a stx_image call.
     
     Exports image data in long-format xyz format for better compatibility.
     Also saves channel data for RGB/RGBA images.
@@ -20,7 +20,7 @@ def _format_plot_image(id, tracked_dict, kwargs):
     Args:
         id (str or int): Identifier for the plot
         tracked_dict (dict): Dictionary containing tracked data
-        kwargs (dict): Keyword arguments passed to plot_image
+        kwargs (dict): Keyword arguments passed to stx_image
         
     Returns:
         pd.DataFrame: Formatted image data in xyz format

scitex/plt/_subplots/_export_as_csv_formatters/{_format_plot_joyplot.py → _format_stx_joyplot.py}

@@ -13,12 +13,12 @@ import pandas as pd
 from scitex.pd import force_df
 
 def _format_plot_joyplot(id, tracked_dict, kwargs):
-    """Format data from a plot_joyplot call.
+    """Format data from a stx_joyplot call.
     
     Args:
         id (str): Identifier for the plot
         tracked_dict (dict): Dictionary containing 'joyplot_data' key with joyplot data
-        kwargs (dict): Keyword arguments passed to plot_joyplot
+        kwargs (dict): Keyword arguments passed to stx_joyplot
         
     Returns:
         pd.DataFrame: Formatted joyplot data

scitex/plt/_subplots/_export_as_csv_formatters/{_format_plot_line.py → _format_stx_line.py}

@@ -11,14 +11,14 @@ __DIR__ = os.path.dirname(__FILE__)
 import pandas as pd
 
 def _format_plot_line(id, tracked_dict, kwargs):
-    """Format data from a plot_line call.
+    """Format data from a stx_line call.
     
-    Processes plot_line data for CSV export.
+    Processes stx_line data for CSV export.
     
     Args:
         id (str): Identifier for the plot
         tracked_dict (dict): Dictionary containing 'plot_df' key with plot data
-        kwargs (dict): Keyword arguments passed to plot_line
+        kwargs (dict): Keyword arguments passed to stx_line
         
     Returns:
         pd.DataFrame: Formatted line plot data

scitex/plt/_subplots/_export_as_csv_formatters/{_format_plot_mean_ci.py → _format_stx_mean_ci.py}

@@ -11,14 +11,14 @@ __DIR__ = os.path.dirname(__FILE__)
 import pandas as pd
 
 def _format_plot_mean_ci(id, tracked_dict, kwargs):
-    """Format data from a plot_mean_ci call.
+    """Format data from a stx_mean_ci call.
     
     Processes mean with confidence interval band plot data for CSV export.
     
     Args:
         id (str): Identifier for the plot
         tracked_dict (dict): Contains 'plot_df' (pandas DataFrame with mean and CI data)
-        kwargs (dict): Keyword arguments passed to plot_mean_ci
+        kwargs (dict): Keyword arguments passed to stx_mean_ci
         
     Returns:
         pd.DataFrame: Formatted mean and CI data

scitex/plt/_subplots/_export_as_csv_formatters/{_format_plot_mean_std.py → _format_stx_mean_std.py}

@@ -11,14 +11,14 @@ __DIR__ = os.path.dirname(__FILE__)
 import pandas as pd
 
 def _format_plot_mean_std(id, tracked_dict, kwargs):
-    """Format data from a plot_mean_std call.
+    """Format data from a stx_mean_std call.
     
     Processes mean with standard deviation band plot data for CSV export.
     
     Args:
         id (str): Identifier for the plot
         tracked_dict (dict): Dictionary containing 'plot_df' key with mean and std data
-        kwargs (dict): Keyword arguments passed to plot_mean_std
+        kwargs (dict): Keyword arguments passed to stx_mean_std
         
     Returns:
         pd.DataFrame: Formatted mean and std data

scitex/plt/_subplots/_export_as_csv_formatters/{_format_plot_median_iqr.py → _format_stx_median_iqr.py}

@@ -11,14 +11,14 @@ __DIR__ = os.path.dirname(__FILE__)
 import pandas as pd
 
 def _format_plot_median_iqr(id, tracked_dict, kwargs):
-    """Format data from a plot_median_iqr call.
+    """Format data from a stx_median_iqr call.
     
     Processes median with interquartile range band plot data for CSV export.
     
     Args:
         id (str): Identifier for the plot
         tracked_dict (dict): Contains 'plot_df' (pandas DataFrame with median and IQR data)
-        kwargs (dict): Keyword arguments passed to plot_median_iqr
+        kwargs (dict): Keyword arguments passed to stx_median_iqr
         
     Returns:
         pd.DataFrame: Formatted median and IQR data

scitex/plt/_subplots/_export_as_csv_formatters/{_format_plot_raster.py → _format_stx_raster.py}

@@ -11,12 +11,12 @@ __DIR__ = os.path.dirname(__FILE__)
 import pandas as pd
 
 def _format_plot_raster(id, tracked_dict, kwargs):
-    """Format data from a plot_raster call.
+    """Format data from a stx_raster call.
     
     Args:
         id (str): Identifier for the plot
         tracked_dict (dict): Dictionary containing 'raster_digit_df' key with raster plot data
-        kwargs (dict): Keyword arguments passed to plot_raster
+        kwargs (dict): Keyword arguments passed to stx_raster
         
     Returns:
         pd.DataFrame: Formatted raster plot data

scitex/plt/_subplots/_export_as_csv_formatters/{_format_plot_rectangle.py → _format_stx_rectangle.py}

@@ -12,7 +12,7 @@ import numpy as np
 import pandas as pd
 
 def _format_plot_rectangle(id, tracked_dict, kwargs):
-    """Format data from a plot_rectangle call."""
+    """Format data from a stx_rectangle call."""
     # Check if tracked_dict is empty or not a dictionary
     if not tracked_dict or not isinstance(tracked_dict, dict):
         return pd.DataFrame()

scitex/plt/_subplots/_export_as_csv_formatters/{_format_plot_scatter_hist.py → _format_stx_scatter_hist.py}

@@ -12,12 +12,12 @@ import numpy as np
 import pandas as pd
 
 def _format_plot_scatter_hist(id, tracked_dict, kwargs):
-    """Format data from a plot_scatter_hist call.
+    """Format data from a stx_scatter_hist call.
     
     Args:
         id (str): Identifier for the plot
         tracked_dict (dict): Dictionary containing tracked data
-        kwargs (dict): Keyword arguments passed to plot_scatter_hist
+        kwargs (dict): Keyword arguments passed to stx_scatter_hist
         
     Returns:
         pd.DataFrame: Formatted scatter histogram data

scitex/plt/_subplots/_export_as_csv_formatters/{_format_plot_shaded_line.py → _format_stx_shaded_line.py}

@@ -11,12 +11,12 @@ __DIR__ = os.path.dirname(__FILE__)
 import pandas as pd
 
 def _format_plot_shaded_line(id, tracked_dict, kwargs):
-    """Format data from a plot_shaded_line call.
+    """Format data from a stx_shaded_line call.
     
     Args:
         id (str): Identifier for the plot
         tracked_dict (dict): Dictionary containing tracked data
-        kwargs (dict): Keyword arguments passed to plot_shaded_line
+        kwargs (dict): Keyword arguments passed to stx_shaded_line
         
     Returns:
         pd.DataFrame: Formatted shaded line data

scitex/plt/_subplots/_export_as_csv_formatters/{_format_plot_violin.py → _format_stx_violin.py}

@@ -12,14 +12,14 @@ import numpy as np
 import pandas as pd
 
 def _format_plot_violin(id, tracked_dict, kwargs):
-    """Format data from a plot_violin call.
+    """Format data from a stx_violin call.
     
     Formats data in a long-format for better compatibility.
     
     Args:
         id (str): Identifier for the plot
         tracked_dict (dict): Dictionary containing tracked data
-        kwargs (dict): Keyword arguments passed to plot_violin
+        kwargs (dict): Keyword arguments passed to stx_violin
         
     Returns:
         pd.DataFrame: Formatted violin plot data in long format