passing multiple callbacks as list

12 months ago · fd08a1f36e
1 changed files with 1 additions and 1 deletions
--- a/examples/shields/rl/15_train_eval_tune.py
+++ b/examples/shields/rl/15_train_eval_tune.py
@ -78,7 +78,7 @@ def ppo(args):
                                  "shielding": args.shielding is ShieldingConfig.Full or args.shielding is ShieldingConfig.Training,
                                  },)
        .framework("torch")
-        .callbacks(MyCallbacks, ShieldInfoCallback)
+        .callbacks([MyCallbacks, ShieldInfoCallback])
        .evaluation(evaluation_config={
                                       "evaluation_interval": 1,
                                        "evaluation_duration": 10,