Update weights.py

Так будет лучше.