Switch to to_numeric in normalize_gpu_stream_numbers (#296)

Summary:
Pull Request resolved: #296

Saves nearly 2 seconds with equivalent output

Reviewed By: A-Kokolis

Differential Revision: D82760298

fbshipit-source-id: 254aa99b84f1aefd431f0051433ee0d70b6f347a

Author: benghaem

hta/utils/utils.py

@@ -5,7 +5,7 @@
 import multiprocessing as mp
 from enum import Enum
 from pathlib import Path
-from typing import Any, List, Tuple
+from typing import List, Tuple
 
 import pandas as pd
 import psutil
@@ -228,16 +228,7 @@ def normalize_gpu_stream_numbers(df: pd.DataFrame) -> None:
         logger.error("No stream column found in the trace.")
         return
 
-    def _normalize_stream_number(stream_number: Any) -> int:
-        try:
-            return int(stream_number)
-        except ValueError:
-            return -1
-
-    df["stream"] = df.apply(
-        lambda r: _normalize_stream_number(r["stream"]),
-        axis=1,
-    )
+    df["stream"] = pd.to_numeric(df["stream"], errors="coerce").fillna(-1).astype(int)
 
 
 def get_value_from_dict(d: object, key: str, default: object = None) -> object: