From ca6cbe8989b066c7733751201783d7ff1683eb5d Mon Sep 17 00:00:00 2001
From: RacerCodeQualityPyreMissingAnnotations Bot
 <noreply+1401739305002972@fb.com>
Date: Thu, 21 May 2026 10:34:49 -0700
Subject: [PATCH] Fix PYRE_MISSING_ANNOTATIONS issues in
 fbcode/HolisticTraceAnalysis/hta/common/execution_trace.py

Reviewed By: fengxizhou

Differential Revision: D105820356
---
 hta/common/execution_trace.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/hta/common/execution_trace.py b/hta/common/execution_trace.py
index 17ab9b20..1e9f4d25 100644
--- a/hta/common/execution_trace.py
+++ b/hta/common/execution_trace.py
@@ -162,7 +162,7 @@ def add_et_column(trace_df: pd.DataFrame, et: ExecutionTrace, column: str) -> No
         return
     if column == "op_schema":
 
-        def map_func(node_id):
+        def map_func(node_id) -> str:
             return et.nodes[node_id].op_schema
 
     elif column == "input_shapes":
@@ -182,7 +182,7 @@ def map_func(node_id):
 
     elif column == "output_types":
 
-        def map_func(node_id):
+        def map_func(node_id) -> List[str]:
             return et.nodes[node_id].output_types
 
     elif column == "et_node_name":