Fix failing test

Author: RaghuSpaceRajan

mdp_playground/spaces/image_multi_discrete.py

@@ -291,8 +291,8 @@ def sample(self):
     def __repr__(self):
         return (
             "{} with multi-discrete space of shape: {} and "
-            "images of resolution: {} x {} and dtype: {}".format(
-                self.__class__, self.state_space_sizes, self.width, self.height, self.dtype
+            "images of resolution: {} and dtype: {}".format(
+                self.__class__, self.state_space_sizes, self.shape, self.dtype
             )
         )
 

mdp_playground/spaces/test_image_multi_discrete.py

@@ -18,30 +18,34 @@ def test_image_multi_discrete(self):
         ds4 = [ds4.n]
         print(ds4)
         imd = ImageMultiDiscrete(ds4, transforms="shift")
+        print("imd.dtype, shape", imd.dtype, imd.shape)
+        print("imd", imd)
         from PIL import Image
 
         # img1 = Image.fromarray(imd.disjoint_states[0][1], 'L')
         img1 = Image.fromarray(np.squeeze(imd.get_concatenated_image(3)), "L")
         if render:
             img1.show()
 
-        imd = ImageMultiDiscrete(
-            ds4,
-            transforms="scale,shift,rotate,flip",
-            use_custom_images="textures",
-            cust_path="/home/rajanr/textures",
-        )
-        img1 = Image.fromarray(np.squeeze(imd.get_concatenated_image(2)), "RGB")
+        if render:
+            imd = ImageMultiDiscrete(
+                ds4,
+                transforms="scale,shift,rotate,flip",
+                use_custom_images="textures",
+                cust_path="/home/rajanr/textures",
+            )
+            img1 = Image.fromarray(np.squeeze(imd.get_concatenated_image(2)), "RGB")
         if render:
             img1.show()
 
-        imd = ImageMultiDiscrete(
-            ds4,
-            transforms="scale,shift,rotate,flip",
-            use_custom_images="images",
-            cust_path="/home/rajanr/textures",
-        )
-        img1 = Image.fromarray(np.squeeze(imd.get_concatenated_image(1)), "RGB")
+        if render:
+            imd = ImageMultiDiscrete(
+                ds4,
+                transforms="scale,shift,rotate,flip",
+                use_custom_images="images",
+                cust_path="/home/rajanr/textures",
+            )
+            img1 = Image.fromarray(np.squeeze(imd.get_concatenated_image(1)), "RGB")
         if render:
             img1.show()