From fd08a1f36e4df270ee2d06e7a3fc5d21a26b11cf Mon Sep 17 00:00:00 2001 From: sp Date: Sat, 30 Dec 2023 11:48:40 +0100 Subject: [PATCH] passing multiple callbacks as list --- examples/shields/rl/15_train_eval_tune.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/shields/rl/15_train_eval_tune.py b/examples/shields/rl/15_train_eval_tune.py index 229b49f..b9abc5e 100644 --- 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,