Error in train.py

[usoylu2]

When i run the following

./pruned_transducer_stateless7_context/train.py \
  --world-size 8 \
  --num-epochs 20 \
  --start-epoch 1 \
  --exp-dir pruned_transducer_stateless7_context/exp \
  --full-libri 1 \
  --max-duration 300

Here is the error:

-- Process 1 terminated with the following error:
Traceback (most recent call last):
File "/local/home/ufsoylu/my_env/lib64/python3.7/site-packages/torch/multiprocessing/spawn.py", line 69, in _wrap
fn(i, *args)
File "/local/home/ufsoylu/icefall_context/icefall/egs/librispeech/ASR/pruned_transducer_stateless7_context/train.py", line 1308, in run
params=params,
File "/local/home/ufsoylu/icefall_context/icefall/egs/librispeech/ASR/pruned_transducer_stateless7_context/train.py", line 1440, in scan_pessimistic_batches_for_oom
is_training=True,
File "/local/home/ufsoylu/icefall_context/icefall/egs/librispeech/ASR/pruned_transducer_stateless7_context/train.py", line 812, in compute_loss
lm_scale=params.lm_scale,
File "/local/home/ufsoylu/my_env/lib64/python3.7/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/local/home/ufsoylu/my_env/lib64/python3.7/site-packages/torch/nn/parallel/distributed.py", line 1040, in forward
output = self._run_ddp_forward(*inputs, **kwargs)
File "/local/home/ufsoylu/my_env/lib64/python3.7/site-packages/torch/nn/parallel/distributed.py", line 1000, in _run_ddp_forward
return module_to_run(*inputs[0], **kwargs[0])
File "/local/home/ufsoylu/my_env/lib64/python3.7/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
TypeError: forward() got an unexpected keyword argument 'word_list'