ensure TDQN class is properly named

This commit is contained in:
robcaulk 2022-08-15 10:29:33 +02:00
parent 91683e1dca
commit d4db5c3281

View File

@ -15,7 +15,7 @@ from stable_baselines3.common.buffers import ReplayBuffer
logger = logging.getLogger(__name__)
class ReinforcementLearningPPO(BaseReinforcementLearningModel):
class ReinforcementLearningTDQN(BaseReinforcementLearningModel):
"""
User created Reinforcement Learning Model prediction model.
"""