If you want to know if you are a resident or non-resident for 2025, that will depend on your immigration and green card status, and how long you were in the US. You might also be considered a dual-status alien for 2025, if you were a resident for part of the year. There are special tax rules for the year you leave the US, and I don't know those rules but other people do.
I thought you were asking about your 401k tax situation. Just having a 401k is not taxed. Withdrawals are taxed, in the year you make them, no matter where you live when you make the withdrawal. Whether you report the income and pay the tax on a 1040 resident return or a 1040NR non-resident return will depend on your status as discussed in the first paragraph. But if you were working in the US when the contributions were made (by you or your employer) then your 401k withdrawals will always be taxed by the US.
