[CI] Local import of PIL

ghstack-source-id: 6eb4ace
Pull Request resolved: #2720

Author: Vincent Moens

test/test_libs.py

@@ -636,7 +636,8 @@ def test_torchrl_to_gym(self, backend, numpy):
 
     @implement_for("gym", None, "0.26")
     def test_gym_dict_action_space(self):
-        pytest.skip("tested for gym > 0.26 - no backward issue")
+        torchrl_logger.info("tested for gym > 0.26 - no backward issue")
+        return
 
     @implement_for("gym", "0.26", None)
     def test_gym_dict_action_space(self):  # noqa: F811

torchrl/envs/custom/chess.py

@@ -10,7 +10,6 @@
 from typing import Dict
 
 import torch
-from PIL import Image
 from tensordict import TensorDict, TensorDictBase
 from torchrl.data import Binary, Bounded, Categorical, Composite, NonTensor, Unbounded
 
@@ -430,6 +429,8 @@ def _torchvision(cls):
     @classmethod
     def _get_tensor_image(cls, board):
         try:
+            from PIL import Image
+
             svg = board._repr_svg_()
             # Convert SVG to PNG using cairosvg
             png_data = io.BytesIO()
@@ -440,7 +441,7 @@ def _get_tensor_image(cls, board):
             img = cls._torchvision.transforms.functional.pil_to_tensor(img)
         except ImportError:
             raise ImportError(
-                "Chess rendering requires cairosvg and torchvision to be installed."
+                "Chess rendering requires cairosvg, PIL and torchvision to be installed."
             )
         return img