AITAH for waiting until I’ve completely recovered before traveling to see my partner after giving birth?

I moved states to be with my partner and had planned to give birth in his home state. However, after some arguments, I decided it was best for me and the baby to return to my home state for the birth. So, at 39 weeks, I drove back home.

I called my partner when I was halfway home and told him that my mental health was really bad, so I decided to return home. He said he supported my decision (this was on a Thursday).

On Sunday night, I went to the hospital because I had reduced fetal movements (I was 39+4). The doctor recommended an induction. I called my partner to update him, and though I don’t remember the details of the conversation, he was very unhappy and angry, which upset me greatly. I was scheduled for an induction on Tuesday.

On Monday morning, my waters broke, so I texted and called my partner. We spoke on the phone, and he said he was torn and uncertain about coming; he was very angry and eventually hung up on me. I went to the hospital and was in active labour when he called and began saying hurtful things, even calling me a "C U Next Tuesday." He spent most of the call ranting while I was in a lot of pain.

He missed the birth but arrived later Monday night (he drove 8 hours to the hospital in my home state). I was very happy to see him, and he was happy to meet his son.

I’ve been staying at my mum's house, and he stayed for a couple of nights before leaving on Thursday, saying he needed to prepare for work on Monday.

He now wants me to return next week. I hemorrhaged during birth and lost nearly a litre of blood, and I had an episiotomy, so I’m still in pain. I informed him that, because of this, I likely won’t be able to return as soon as he wants.

In our conversation tonight over the phone, he said he doesn’t want to miss out on six weeks of his son’s life (or however long it takes me to recover). He mentioned buying a caravan or a train ticket with a bed so I could lie down while traveling. Alternatively, he suggested I could fly for an hour and a half. He said he left early because he didn’t want to stay at my mum's house.

He wants me to make the sacrifice of spending an hour and a half on a plane to return home, claiming it would be a win-win situation. He said that if I can endure 10 hours of labour, I should be able to manage an hour and a half on a plane. He believes I should make this small sacrifice and that he has made sacrifices, so I should be able to make one too.

AITAH for waiting until I’ve fully recovered from giving birth and feel more comfortable with the baby and breastfeeding before returning?