fixt test

snowflakedb · Jun 13, 2024 · 80517fc · 80517fc
1 parent cb68452
commit 80517fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/snowflake/snowpark/_internal/open_telemetry.py b/src/snowflake/snowpark/_internal/open_telemetry.py
@@ -137,8 +137,8 @@ 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:
+            target_index = i + decorator_number + 1
             if file_name in registration_class:
                 continue
             break