[SPARK-38556] Disable Pandas usage logging for method calls inside @contextmanager functions

• Type: Improvement
• Status: In Progress
• Priority: Minor
• Resolution: Unresolved
• Affects Version/s: 3.2.1
• Component/s: PySpark
• Target Version/s: 3.2.1, 3.3.0
• Assignee: Unassigned
• Reporter: Yihong He
• Shepherd: Takuya Ueshin
• Created: 15/Mar/22 15:48
• Updated: 12/Dec/22 18:10

Description

Currently, calls inside @contextmanager functions are treated as external for with statements.

For example, the below code records config.set_option calls inside ps.option_context(...)

with ps.option_context("compute.ops_on_diff_frames", True):
    pass 

We should disable usage logging for calls inside @contextmanager functions to improve accuracy of the usage data

 

Issue Links

• links to: [Github] Pull Request #35861 (heyihong)