From cb68452acc38ae572fa451bdfadd84cab8c766f9 Mon Sep 17 00:00:00 2001 From: Yuyang Wang Date: Thu, 13 Jun 2024 14:06:21 -0700 Subject: [PATCH] fix test --- src/snowflake/snowpark/_internal/open_telemetry.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/snowflake/snowpark/_internal/open_telemetry.py b/src/snowflake/snowpark/_internal/open_telemetry.py index a3612bc2991..8e29476ac68 100644 --- a/src/snowflake/snowpark/_internal/open_telemetry.py +++ b/src/snowflake/snowpark/_internal/open_telemetry.py @@ -137,11 +137,10 @@ def context_manager_code_location(frame_info, func) -> Tuple[str, int]: target_index = -1 for i, frame in enumerate(frame_info): file_name = os.path.basename(frame.filename) + target_index = i + decorator_number + 1 if file_name in target_class: if file_name in registration_class: - target_index = i + decorator_number + 1 continue - target_index = i + decorator_number + 1 break frame = frame_info[target_index] return frame.filename, frame.lineno