You're not wrong! I think it is Yaz's iPhone 6+ - here's a photo of the Doctor holding it: Doctor holding Yaz's phone. Later in the episode, she asks Yaz for it, before remembering it's in her own pocket.