From cd81bcc9787660757e6a27b89a0d58590c02b791 Mon Sep 17 00:00:00 2001 From: Varun Sundar Rabindranath Date: Tue, 9 Jul 2024 18:53:11 +0000 Subject: [PATCH] fix plot y label --- neuralmagic/tools/profiler/visualize_trace.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/neuralmagic/tools/profiler/visualize_trace.py b/neuralmagic/tools/profiler/visualize_trace.py index b45d0f83fdbf4..8d9ff24916b8a 100644 --- a/neuralmagic/tools/profiler/visualize_trace.py +++ b/neuralmagic/tools/profiler/visualize_trace.py @@ -268,7 +268,7 @@ def plot_trace_df(traces_df: pd.DataFrame, # Setup labels and title plt.setp(ax.get_xticklabels(), rotation=90) - ax.set_ylabel("time ms") + ax.set_ylabel(plot_metric) plt.suptitle(plot_title) plt.savefig(output, bbox_inches='tight')