Checking onboardin flow - fix 4

2025-07-27 16:29:53 +02:00
parent 0b14cf9eb2
commit e63a99b818
8 changed files with 497 additions and 8 deletions
--- a/services/training/app/ml/trainer.py
+++ b/services/training/app/ml/trainer.py
@@ -244,6 +244,10 @@ class BakeryMLTrainer:
        if sales_df.empty:
            raise ValueError(f"No sales data provided for tenant {tenant_id}")
        
+        if 'quantity_sold' in sales_df.columns and 'quantity' not in sales_df.columns:
+            sales_df['quantity'] = sales_df['quantity_sold']
+            logger.info("Mapped 'quantity_sold' to 'quantity' column")
+        
        required_columns = ['date', 'product_name', 'quantity']
        missing_columns = [col for col in required_columns if col not in sales_df.columns]
        if missing_columns: