From daa1480a910d0b9a803b7b5d1b51dc1443a7553c Mon Sep 17 00:00:00 2001 From: Michel Aractingi Date: Wed, 22 Jan 2025 09:45:05 +0100 Subject: [PATCH] nit --- lerobot/scripts/eval_on_robot.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/scripts/eval_on_robot.py b/lerobot/scripts/eval_on_robot.py index 66ccc605..6fb79c0a 100644 --- a/lerobot/scripts/eval_on_robot.py +++ b/lerobot/scripts/eval_on_robot.py @@ -246,7 +246,7 @@ def eval_policy( start_eval = time.perf_counter() progbar = trange(n_episodes, desc="Evaluating policy on real robot") - reward_classifier = get_classifier(reward_classifier_pretrained_path, reward_classifier_config_file).to(device)ยง + reward_classifier = get_classifier(reward_classifier_pretrained_path, reward_classifier_config_file).to(device) device = get_device_from_parameters(policy) if device is None else device