Skip to content

Commit

Permalink
Fix broken test
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 549289005
Change-Id: Ie90fa2ddbd7d5d90f969d618126c20e287533348
  • Loading branch information
jagapiou authored and Copybara-Service committed Jul 19, 2023
1 parent 288722a commit 9b89c37
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions examples/rllib/utils_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
from absl.testing import absltest
from gym.spaces import discrete
from meltingpot import substrate
from meltingpot.configs.substrates import commons_harvest_open
from meltingpot.configs.substrates import commons_harvest__open

from . import utils

Expand All @@ -27,12 +27,12 @@ class MeltingPotEnvTests(absltest.TestCase):
def setUp(self):
super().setUp()
# Create a new MeltingPotEnv for each test case
self._env_config = substrate.get_config('commons_harvest_open')
self._env_config = substrate.get_config('commons_harvest__open')
self._env = utils.env_creator(self._env_config)

def test_action_space_size(self):
"""Test the action space is the correct size."""
actions_count = len(commons_harvest_open.ACTION_SET)
actions_count = len(commons_harvest__open.ACTION_SET)
env_action_space = self._env.action_space['player_1']
self.assertEqual(env_action_space, discrete.Discrete(actions_count))

Expand Down

0 comments on commit 9b89c37

Please sign in to comment.