Реестр агентов A2C

Question

Я думаю, что в файле ray/rllib/agent /registry.py для регистрации A2C Trainer есть ошибка. В данный момент:

def _import_a2c():
    from ray.rllib.agents import a3c
    return a3c.A2CTrainer

В файле a3c.py нет переменной A2CTrainer, но в файле a2c.py есть.

Думаю, должно быть:

def _import_a2c():
    from ray.rllib.agents import a2c
    return a2c.A2CTrainer