Any gains from an investment sale are always included as state income. Your Sprintax agent should be able to assist you with the state return as well as the federal return, Form 1040NR.
https://www.sprintax.com/non-resident-alien-tax-1040nr-turbotax.html
