Skip to content

⚡ Bolt: [performance improvement] Optimize DataFrame iterrows in ETL pipeline#8

Open
Vagarh wants to merge 1 commit into
masterfrom
bolt-etl-iterrows-optimization-4263350885591088969
Open

⚡ Bolt: [performance improvement] Optimize DataFrame iterrows in ETL pipeline#8
Vagarh wants to merge 1 commit into
masterfrom
bolt-etl-iterrows-optimization-4263350885591088969

Conversation

@Vagarh

@Vagarh Vagarh commented Jul 11, 2026

Copy link
Copy Markdown
Owner

💡 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 the load_data task of e2e_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 the load_task operator in the Airflow UI compared to previous runs.


PR created automatically by Jules for task 4263350885591088969 started by @Vagarh

Co-authored-by: Vagarh <111590756+Vagarh@users.noreply.github.com>
@google-labs-jules

Copy link
Copy Markdown
Contributor

👋 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 @jules. You can find this option in the Pull Request section of your global Jules UI settings. You can always switch back!

New to Jules? Learn more at jules.google/docs.


For security, I will only act on instructions from the user who triggered this task.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant