Logistic Regression - Pytorch Typing CST Test