Diatta pelasi Senegalin maajoukkueessa ensimmäisen maaottelunsa maaliskuussa 2019 Madagaskaria vastaan. Ensimmäisen maaottelumaalinsa hän teki kesäkuussa 2019 Afrikan-mestaruuskilpailujen alkulohko-ottelussa Tansaniaa vastaan. Hän on pelannut kaikkiaan kymmenen maaottelua.