Skip to content

Commit

Permalink
lint changes
Browse files Browse the repository at this point in the history
  • Loading branch information
sriramsk1999 committed Apr 7, 2024
1 parent 7ee8790 commit ba59958
Showing 1 changed file with 4 additions and 9 deletions.
13 changes: 4 additions & 9 deletions test/test_libs.py
Original file line number Diff line number Diff line change
Expand Up @@ -3339,22 +3339,16 @@ class TestRoboHive:
# Locally these imports can be annoying, especially given the amount of
# stuff printed by robohive.
@pytest.mark.parametrize("from_pixels", [True, False])
@pytest.mark.parametrize("envname", RoboHiveEnv.available_envs)
@pytest.mark.parametrize("envname", RoboHiveEnv.available_envs[:10])
@set_gym_backend("gym")
def test_robohive(self, envname, from_pixels):
if any(
substr in envname
for substr in ("_vr3m", "_vrrl", "_vflat", "_vvc1s")
):
if any(substr in envname for substr in ("_vr3m", "_vrrl", "_vflat", "_vvc1s")):
torchrl_logger.info("not testing envs with prebuilt rendering")
return
if "Adroit" in envname:
torchrl_logger.info("tcdm are broken")
return
if (
from_pixels
and len(RoboHiveEnv.get_available_cams(env_name=envname)) == 0
):
if from_pixels and len(RoboHiveEnv.get_available_cams(env_name=envname)) == 0:
torchrl_logger.info("no camera")
return
try:
Expand All @@ -3367,6 +3361,7 @@ def test_robohive(self, envname, from_pixels):
raise err
check_env_specs(env)


@pytest.mark.skipif(not _has_smacv2, reason="SMACv2 not found")
class TestSmacv2:
def test_env_procedural(self):
Expand Down

0 comments on commit ba59958

Please sign in to comment.