Fix sweep bug

PufferAI · Feb 17, 2025 · d13630e · d13630e
1 parent c327d92
commit d13630e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo.py b/demo.py
@@ -89,7 +89,7 @@ def sweep(args, env_name, make_env, policy_cls, rnn_cls):
         np.random.seed(seed)
         torch.manual_seed(seed)
 
-        info = sweep.suggest(args['sweep'])
+        info = sweep.suggest(args)
         score, cost, _, _ = train(args, make_env, policy_cls, rnn_cls, target_metric)
         sweep.observe(score, cost)
         print('Score:', score, 'Cost:', cost)