From b1594260ba6b3ac15dc18f0ef6d899697579c293 Mon Sep 17 00:00:00 2001 From: Artur Niederfahrenhorst Date: Mon, 25 Jul 2022 15:12:42 +0200 Subject: [PATCH] [RLlib] Small SlateQ example fix. (#26948) --- rllib/examples/recommender_system_with_recsim_and_slateq.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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))