@sudimax , since for 2020 you passed the Substantial Presence Test, before leaving the states and you had no income for the rest of the year, the easiest path would be to file a form 1040 on a short year -- I am assuming here that you had ZERO income from the USA while you were gone ( expiry of your visa at departure ). TurboTax can be sued to file form 1040 as a resident for the year.
For 2021, since you are not an exempt any more , you would be counting days present towards SPT
