traveling to India with formula fed baby?

Hello! We are leaving to India next month and our baby is formula fed. She is 4 months old. We are thinking of buying similac on-the-go bottles for her for lesser hassle with mixing and feeding and washing 😑... I've just gone through TSA website and they didn't mention any restrictions on the amount of formula allowed... which was a relief. How ever, I dunno if the customs check at India would be the same... are there any mom's here who have been to India and back to US with formula fed baby? If so, can you give me any advice or tips? Thank you!