"Looking at your error, it appears get_num_classes doesn't exist anymore. I verified this by looking that their github and docs. It was removed after this commit."