⚡ Bolt: [performance improvement] Optimize iterrows in ETL pipeline#10
⚡ Bolt: [performance improvement] Optimize iterrows in ETL pipeline#10Vagarh 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 Pandas DataFrame conversion and
df.iterrows()with a direct loop over a list of dictionaries (data) inside theload_datafunction inpublic_data_etl.py.🎯 Why: Converting a perfectly good native list of dictionaries into a Pandas DataFrame solely to iterate over its rows using
iterrows()is a massive anti-pattern in Airflow pipelines.iterrows()creates a Pandas Series for every single row, causing huge memory and O(n) CPU overhead, while a simple pythonforloop is lightning-fast and requires no extra memory.📊 Impact: Massively reduces CPU and memory utilization during the
load_dataDAG step, which is particularly beneficial as the dataset size grows or Airflow worker memory is constrained.🔬 Measurement: Local profiling would show almost zero memory allocation during the tuple preparation block and vastly faster completion times compared to iterating via DataFrames.
PR created automatically by Jules for task 11047276891903090654 started by @Vagarh