Warning quando executando código classifica_busca.py

Boa noite.

Ao executar o código da aula 03 (Vídeo 7), foi exibido o seguinte resultado:

Air-de-Duany:alura-ia duany$ python classifica_buscas.py 
classifica_buscas.py:19: VisibleDeprecationWarning: using a non-integer number instead of an integer will result in an error in the future
  treino_dados = X[:tamanho_de_treino]
classifica_buscas.py:20: VisibleDeprecationWarning: using a non-integer number instead of an integer will result in an error in the future
  treino_marcacoes = Y[:tamanho_de_treino]
classifica_buscas.py:22: VisibleDeprecationWarning: using a non-integer number instead of an integer will result in an error in the future
  teste_dados = X[-tamanho_de_teste:]
classifica_buscas.py:23: VisibleDeprecationWarning: using a non-integer number instead of an integer will result in an error in the future
  teste_marcacoes = Y[-tamanho_de_teste:]
82.0
100

Obtive 82% de acerto no conjunto de testes. Entretanto, minha dúvida refere-se ao warning exibido. Creio que seja em virtude do valor das variáveis tamanho_de_treino e tamanho_de_teste serem decimais, pois elas foram obtidas da seguinte forma:

porcentagem_de_treino = 0.9
tamanho_de_treino = porcentagem_de_treino * len(Y)
tamanho_de_teste = len(Y) - tamanho_de_treino

Acredito que esse warning não ocorreu no vídeo da Aula em razão da versão do ambiente python que utilizo (Python versão 3.5).

Gostaria de saber como proceder para resolver isso da forma mais elegante. Posso apenas utilizar a função int() na declaração das variáveis tamanho_de_treino e tamanho_de_teste, como no exemplo abaixo?

porcentagem_de_treino = 0.9
tamanho_de_treino = int(porcentagem_de_treino * len(Y))
tamanho_de_teste = int(len(Y) - tamanho_de_treino)

Abraços.