⚡ Bolt: [performance improvement] Optimize DataFrame iterrows in ETL pipeline#8
⚡ Bolt: [performance improvement] Optimize DataFrame iterrows in ETL pipeline#8Vagarh wants to merge 1 commit into
Conversation
Co-authored-by: Vagarh <111590756+Vagarh@users.noreply.github.com>
|
👋 Jules, reporting for duty! I'm here to lend a hand with this pull request. When you start a review, I'll add a 👀 emoji to each comment to let you know I've read it. I'll focus on feedback directed at me and will do my best to stay out of conversations between you and other bots or reviewers to keep the noise down. I'll push a commit with your requested changes shortly after. Please note there might be a delay between these steps, but rest assured I'm on the job! For more direct control, you can switch me to Reactive Mode. When this mode is on, I will only act on comments where you specifically mention me with New to Jules? Learn more at jules.google/docs. For security, I will only act on instructions from the user who triggered this task. |
💡 What:
Replaced the creation of a pandas DataFrame and iteration using
df.iterrows()with a direct iteration over the native Python list of dictionaries in theload_datatask ofe2e_open_data_pipeline/dags/public_data_etl.py.🎯 Why:
The ETL task pulled a JSON list of dictionaries from XCom and converted it to a Pandas DataFrame purely to iterate over its rows using
iterrows()for inserting data into Postgres.iterrows()is notoriously slow, and allocating a DataFrame just to iterate over it introduces unnecessary memory overhead and processing time compared to basic Python collections.📊 Impact:
Significantly reduces memory footprint and CPU processing time during the data loading phase, particularly as the number of records pulled from the API increases. The loop operation is now near-instantaneous native dict traversal instead of costly Pandas Series conversions.
🔬 Measurement:
Run the ETL DAG (
open_data_etl_accidentes) and observe the execution time and memory usage of theload_taskoperator in the Airflow UI compared to previous runs.PR created automatically by Jules for task 4263350885591088969 started by @Vagarh