O valor que é retornado para o campo tamanho_de_treino é 900.0, eu não preciso transformar ele em int para não dar erro ?

Referente ao curso Machine Learning: classificação por trás dos panos, no capítulo Classificação de variáveis categóricas e atividade Pegando os dados e marcações de teste e de treino