From fd08a1f36e4df270ee2d06e7a3fc5d21a26b11cf Mon Sep 17 00:00:00 2001
From: sp <stefan.pranger@iaik.tugraz.at>
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,