В данном случае правильный ответ кроется в общем смысле предложения.
Правильно
По окончании — если мы говорим в контексте «после какого-либо события», тогда в конце слова пишется буква «и».
По окончании уроков мы пошли в актовый зал.
По окончании работы они ушли домой.
По окончании школы всем нужно выбрать профессию.
По окончании первого сезона актеры дали интервью.
По окончанию — если предложение строится в контексте «по признаку», тогда в конце слова пишется буква «ю».
По окончанию слова можно определить падеж.
По окончанию фильма я понял, что это только первая часть.
По окончанию песни я узнал автора.
По окончании фильма мы пошли домой — после того, как закончился фильм, мы пошли домой.
По окончанию фильма мы плакали — концовка фильма была такой, что мы плакали.