TypeError: __init__() got an unexpected keyword argument 'agent' in elegantrl_models.py

I get the following error while running the elegantrl_models.py:

TypeError: __init__() got an unexpected keyword argument 'agent' in the 58th line of /agents/elegantrl_models.py

I understand "agent" argument in Arguments class of config.py has been changed to "agent_class"

So when I change the "agent" argument in the elegantrl_models.py to "agent_class", I get the following error:

ValueError: too many values to unpack (expected 2), in the 107th line of elegantrl/train/run.py