Thomas Partey set to sign new Atletico Madrid contract