from .custom_hooks import TransferWeight