I'd return.
HOWEVER, I know that it's totally against my nature to do that. I'd stress so much about school and striving to be the top of the class that it would make me miserable. I'd ease myself of that burden as soon as possible and leave for the NBA.
Maybe I'd flop in the NBA, but it would be only one thnig to stress about rather than two.