Total of 3 rounds consisting of
2 technical rounds
and 1 non technical round
Non technical round was basically a personality check.
1st technical round had all the coding part covered in it. From python to spark l, sql hive etc.
2nd technical also consisted of the coding skills measuring the optimisation/time complexity of the code written and also the situation based problems and the solutions around it.
Post that the HR discussion.