fixed include

2 years ago · 0f3af93c68
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
@ -18,7 +18,7 @@ from helpers import parse_arguments, create_log_dir, ShieldingConfig, test_name
 from shieldhandlers import MiniGridShieldHandler, create_shield_query
 from torch.utils.tensorboard import SummaryWriter
 from callbacks import MyCallbacks
 from callbacks import MyCallbacks, ShieldInfoCallback
 def shielding_env_creater(config):