When I had to enter only NY wages on on non-resident return, I used the allocation worksheet.  I simply typed in the amount of wages earned in NY in column D.  Total wages are already present in column A if you chose non-resident.  Then client was only taxed on that amount not the total wages. 

However I am from Iowa so maybe someone from NY area would be better off responding.