bakery-admin/bakery-ia
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Start fixing forecast service 17

Browse Source
This commit is contained in:
Urtzi Alfaro
2025-07-30 08:29:40 +02:00
parent e5c3375d53
commit c10a745695
3 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
services/training/app/ml/data_processor.py
View File

@@ -390,7 +390,15 @@ class BakeryDataProcessor:
if 'date' not in traffic_clean.columns and 'ds' in traffic_clean.columns:
traffic_clean = traffic_clean.rename(columns={'ds': 'date'})
# 🔧 FIX: Ensure timezone awareness before merge
traffic_clean['date'] = pd.to_datetime(traffic_clean['date'])
# If timezone-naive, localize to UTC
if traffic_clean['date'].dt.tz is None:
traffic_clean['date'] = traffic_clean['date'].dt.tz_localize('UTC')
# If already timezone-aware but not UTC, convert to UTC
elif str(traffic_clean['date'].dt.tz) != 'UTC':
traffic_clean['date'] = traffic_clean['date'].dt.tz_convert('UTC')
# Map traffic columns to standard names
traffic_mapping = {