Update main.py

main.py

@@ -160,9 +160,8 @@ def forecast(monthly_sales):
 
 def process_product(product_name, data):
     full_trend = ""
-    
-    sub_sku_list = data['sub_sku'].tolist()
-    print(sub_sku_list[0])
+
+    sub_sku = data['sub_sku'].tolist()[0]
     try:
         # Get today's date
         today = datetime.today().strftime('%Y-%m-%d %H:%M:%S')