a bug need to fix

TypeError: avgloss_logger_hook() takes 4 positional arguments but 5 were given

while in easytransfer/engines/model.py
avgloss_hook = avgloss_logger_hook(self.train_steps,
total_loss,
self.model_dir,
self.config.log_step_count_steps,
self.config.task_index)

and in easytransfer/utils/hooks.py
def avgloss_logger_hook(max_steps, loss, model_dir, log_step_count_steps):

I have installed EasyTransfer via pip install

Inspect easytransfer/engines/model.py at the avgloss_logger_hook call and compare it with the definition in easytransfer/utils/hooks.py. Run the affected training path to reproduce the TypeError; done means the hook call and definition agree and training proceeds without this argument-count failure.

Written by the indexing model from the issue text.