Fix type mismatch due to same variable naming

dantp-ai · Jan 3, 2024 · 1468cdf · 1468cdf
1 parent 7404a76
commit 1468cdf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/offline/test_bcq.py b/test/offline/test_bcq.py
@@ -206,8 +206,8 @@ def watch():
         env = gym.make(args.task)
         policy.eval()
         collector = Collector(policy, env)
-        result = collector.collect(n_episode=1, render=args.render)
-        print(f"Final reward: {result.returns_stat.mean}, length: {result.lens_stat.mean}")
+        res = collector.collect(n_episode=1, render=args.render)
+        print(f"Final reward: {res.returns_stat.mean}, length: {res.lens_stat.mean}")
 
 
 if __name__ == "__main__":