Skip to content

Commit dfa5cfd

Browse files
maryamhonariGitHub Enterprise
authored andcommitted
remove torch shadow in trainers tests (#69)
1 parent bd5c8ea commit dfa5cfd

28 files changed

+7
-7
lines changed

ml-agents/mlagents/trainers/tests/torch/saver/test_saver.py renamed to ml-agents/mlagents/trainers/tests/torch_entities/saver/test_saver.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
POCASettings,
2323
)
2424
from mlagents.trainers.tests import mock_brain as mb
25-
from mlagents.trainers.tests.torch.test_policy import create_policy_mock
25+
from mlagents.trainers.tests.torch_entities.test_policy import create_policy_mock
2626
from mlagents.trainers.torch_entities.utils import ModelUtils
2727

2828

ml-agents/mlagents/trainers/tests/torch/saver/test_saver_reward_providers.py renamed to ml-agents/mlagents/trainers/tests/torch_entities/saver/test_saver_reward_providers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@
1818
SACSettings,
1919
POCASettings,
2020
)
21-
from mlagents.trainers.tests.torch.test_policy import create_policy_mock
22-
from mlagents.trainers.tests.torch.test_reward_providers.utils import (
21+
from mlagents.trainers.tests.torch_entities.test_policy import create_policy_mock
22+
from mlagents.trainers.tests.torch_entities.test_reward_providers.utils import (
2323
create_agent_buffer,
2424
)
2525

0 commit comments

Comments
 (0)