diff --git a/rllib/examples/recommender_system_with_recsim_and_slateq.py b/rllib/examples/recommender_system_with_recsim_and_slateq.py index ebe522a4a..59e80ef8e 100644 --- a/rllib/examples/recommender_system_with_recsim_and_slateq.py +++ b/rllib/examples/recommender_system_with_recsim_and_slateq.py @@ -181,7 +181,7 @@ def main(): if args.run == "DQN": trainer = dqn.DQN(config=config) else: - trainer = slateq.SlateQTrainer(config=config) + trainer = slateq.SlateQ(config=config) for i in range(10): result = trainer.train() print(pretty_print(result))