From 37bbb58a19b04020b087b908fe8d2d1ac585cb72 Mon Sep 17 00:00:00 2001 From: jamessiddeley-amd Date: Thu, 6 Nov 2025 11:13:10 -0500 Subject: [PATCH] [rocprof-compute] fix unit regex 'ns' in analyze mode (#1689) * fix unit regex in analyze mode * ruff format --- projects/rocprofiler-compute/src/utils/tty.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/projects/rocprofiler-compute/src/utils/tty.py b/projects/rocprofiler-compute/src/utils/tty.py index ea839762a4..76587b2ef0 100644 --- a/projects/rocprofiler-compute/src/utils/tty.py +++ b/projects/rocprofiler-compute/src/utils/tty.py @@ -97,7 +97,9 @@ def convert_time_columns(df: pd.DataFrame, time_unit: str) -> pd.DataFrame: # Avoid modifying the original df_copy = df.copy() - time_rows = df_copy["Unit"].str.lower().str.contains("ns", na=False) + time_rows = ( + df_copy["Unit"].str.lower().str.contains(r"\bns\b", na=False, regex=True) + ) time_value_columns = ["Avg", "Min", "Max"] for col in time_value_columns: @@ -127,7 +129,9 @@ def has_time_data(df: pd.DataFrame) -> bool: if "Unit" not in df.columns: return False # NOTE: "ns" / "NS" / "nS" / "Ns" are reserved for Nanosec time unit - return bool(df["Unit"].str.lower().str.contains("ns", na=False).any()) + return bool( + df["Unit"].str.lower().str.contains(r"\bns\b", na=False, regex=True).any() + ) def is_roofline_shown(