OMG I am the worst. The labels in the other dataset were numbers as strings instead of ints. Changing to int has worked.