From 2a2f34ba878d09cf3ccc44787f77a6363709d342 Mon Sep 17 00:00:00 2001 From: vincentpierre Date: Wed, 29 Aug 2018 10:20:48 -0700 Subject: [PATCH] Fixed the gym_unity tests --- gym-unity/tests/test_gym.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/gym-unity/tests/test_gym.py b/gym-unity/tests/test_gym.py index eebc19b197..a5149b042b 100644 --- a/gym-unity/tests/test_gym.py +++ b/gym-unity/tests/test_gym.py @@ -5,8 +5,8 @@ from gym_unity.envs import UnityEnv, UnityGymException from tests.mock_communicator import MockCommunicator -@mock.patch('unityagents.UnityEnvironment.executable_launcher') -@mock.patch('unityagents.UnityEnvironment.get_communicator') +@mock.patch('mlagents.envs.UnityEnvironment.executable_launcher') +@mock.patch('mlagents.envs.UnityEnvironment.get_communicator') def test_gym_wrapper(mock_communicator, mock_launcher): mock_communicator.return_value = MockCommunicator( discrete_action=False, visual_inputs=0, stack=False, num_agents=1) @@ -26,8 +26,8 @@ def test_gym_wrapper(mock_communicator, mock_launcher): assert isinstance(done, bool) assert isinstance(info, dict) -@mock.patch('unityagents.UnityEnvironment.executable_launcher') -@mock.patch('unityagents.UnityEnvironment.get_communicator') +@mock.patch('mlagents.envs.UnityEnvironment.executable_launcher') +@mock.patch('mlagents.envs.UnityEnvironment.get_communicator') def test_multi_agent(mock_communicator, mock_launcher): mock_communicator.return_value = MockCommunicator( discrete_action=False, visual_inputs=0, stack=False, num_agents=2)