Post flop CO was getting correct odds to call a nut flush draw. He could also be slow playing AA, dangerous, with the club.
The thing I don't like here is the check on the turn. You should bet out for some info. If you get action, act accordingly.
As for the river, I think a call would have been a better move. If he has the A you're done either way. At least with the call you loose less.