Visite de reprise, inaptitude catégorie 1 et rééducation professionnelle

Je suis actuellement en rééducation professionnelle dans un centre spécialisée. Mon employeur est au courant puisqu'il a dû m'établir un certificat de non rémunération...

Je termine ce stage dans 3 mois, et ai été déclarée en invalidité de 1ère catégorie depuis le 1er avril.

On me dit que j'aurais dû demander à mon employeur une visite de reprise avant la date d'effet de mon invalidité. mais également que maintenant je devrais attendre la fin de mon stage de rééducation pour effectuer cette visite, car mon médecin m'a prescrit un arrêt de travail jusqu'à la fin de ce stage pour justifier de mon absence...

Quelles sont les règles ? Dois je solliciter mon employeur pour effectuer cette reprise à la fin de ce stage, ou est il possible de l'anticiper sachant qu'il y a de fortes chances que je sois déclarée inapte à mon poste, raison pour laquelle je suis en phase de recherche de reconversion ?   

Merci pour vos réponses.