State tax filing

Yes, you can allocate your HSA contributions in your Arizona non-resident return. In fact, Arizona will allow you to allocate your income and expenses as you continue through your Arizona non-resident return. 

To allocate your HSA contribution in the Arizona return, you will allocate in the adjustment screen as you proceed through the state interview. Here is what the screen looks like in your Arizona return in Turbo Tax.