diff --git a/test/test_libs.py b/test/test_libs.py index 7ddb0d4fc02..cfcde85cc2c 100644 --- a/test/test_libs.py +++ b/test/test_libs.py @@ -3416,7 +3416,9 @@ def test_robohive(self, envname, from_pixels, from_depths): torchrl_logger.info("no camera") return try: - env = RoboHiveEnv(envname, from_pixels=from_pixels, from_depths=from_depths) + env = RoboHiveEnv( + envname, from_pixels=from_pixels, from_depths=from_depths + ) except AttributeError as err: if "'MjData' object has no attribute 'get_body_xipos'" in str(err): torchrl_logger.info("tcdm are broken")