You can attend school with pending AOS based on marriage. Be sure to talk to your school, though, so they know what's going. They may want documentation of your petition.