From dd7e4661044387408bed49fd0a63a57e4d5c3dd9 Mon Sep 17 00:00:00 2001 From: Charlie Crane Date: Tue, 6 Aug 2024 15:01:53 +0100 Subject: [PATCH] #2706 - Fixing pipeline failure --- .../action_masking/test_agents_use_action_masks.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/e2e_integration_tests/action_masking/test_agents_use_action_masks.py b/tests/e2e_integration_tests/action_masking/test_agents_use_action_masks.py index 4260c605..addf6dca 100644 --- a/tests/e2e_integration_tests/action_masking/test_agents_use_action_masks.py +++ b/tests/e2e_integration_tests/action_masking/test_agents_use_action_masks.py @@ -1,6 +1,7 @@ # © Crown-owned copyright 2024, Defence Science and Technology Laboratory UK from typing import Dict +import pytest import yaml from ray.rllib.algorithms.ppo import PPOConfig from ray.rllib.core.rl_module.marl_module import MultiAgentRLModuleSpec