From bbca8bd37253bf679d4ec006f0e8987847e5ea0e Mon Sep 17 00:00:00 2001 From: sp Date: Sat, 30 Dec 2023 11:58:12 +0100 Subject: [PATCH] fixed include --- 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 8b29ced..a71dfcc 100644 --- a/examples/shields/rl/15_train_eval_tune.py +++ b/examples/shields/rl/15_train_eval_tune.py @@ -10,7 +10,7 @@ from ray.tune.logger import UnifiedLogger from ray.rllib.models import ModelCatalog from ray.tune.logger import pretty_print, UnifiedLogger, CSVLogger from ray.rllib.algorithms.algorithm import Algorithm -from ray.rllib.algorithm.callbacks import make_multi_callbacks +from ray.rllib.algorithms.callbacks import make_multi_callbacks from ray.air import session from torch_action_mask_model import TorchActionMaskModel