fix: rendering in cvrp and tsp (#251)

MANIFEST.in

@@ -2,6 +2,7 @@ include LICENSE
 include requirements/*
 recursive-include * *.npy
 recursive-include jumanji *.png
+recursive-include jumanji *.jpeg
 
 # remove the test specific files
 recursive-exclude * *_test.py

jumanji/environments/routing/cvrp/viewer.py

@@ -18,6 +18,7 @@
 import matplotlib.animation
 import matplotlib.pyplot as plt
 import numpy as np
+import pkg_resources
 from chex import Array
 from numpy.typing import NDArray
 
@@ -146,7 +147,10 @@ def _prepare_figure(self, ax: plt.Axes) -> None:
         ax.set_ylim(0, 1)
         ax.get_xaxis().set_visible(False)
         ax.get_yaxis().set_visible(False)
-        map_img = plt.imread("docs/img/city_map.jpeg")
+        img_path = pkg_resources.resource_filename(
+            "jumanji", "environments/routing/cvrp/img/city_map.jpeg"
+        )
+        map_img = plt.imread(img_path)
         ax.imshow(map_img, extent=[0, 1, 0, 1])
 
     def _group_tour(self, tour: Array) -> list:

jumanji/environments/routing/multi_cvrp/viewer.py

@@ -19,6 +19,7 @@
 import matplotlib.animation
 import matplotlib.pyplot as plt
 import numpy as np
+import pkg_resources
 from numpy.typing import NDArray
 
 import jumanji.environments
@@ -158,7 +159,11 @@ def _prepare_figure(self, ax: plt.Axes) -> None:
         ax.set_ylim(0, 1)
         ax.get_xaxis().set_visible(False)
         ax.get_yaxis().set_visible(False)
-        map_img = plt.imread("docs/img/city_map.jpeg")
+
+        img_path = pkg_resources.resource_filename(
+            "jumanji", "environments/routing/multi_cvrp/img/city_map.jpeg"
+        )
+        map_img = plt.imread(img_path)
         ax.imshow(map_img, extent=[0, 1, 0, 1])
 
     def _group_tour(self, tour: chex.Array) -> list:

jumanji/environments/routing/tsp/viewer.py

@@ -17,6 +17,7 @@
 import matplotlib.animation
 import matplotlib.pyplot as plt
 import numpy as np
+import pkg_resources
 from numpy.typing import NDArray
 
 import jumanji.environments
@@ -134,7 +135,10 @@ def _prepare_figure(self, ax: plt.Axes) -> None:
         ax.set_ylim(0, 1)
         ax.get_xaxis().set_visible(False)
         ax.get_yaxis().set_visible(False)
-        map_img = plt.imread("docs/img/city_map.jpeg")
+        img_path = pkg_resources.resource_filename(
+            "jumanji", "environments/routing/tsp/img/city_map.jpeg"
+        )
+        map_img = plt.imread(img_path)
         ax.imshow(map_img, extent=[0, 1, 0, 1])
 
     def _add_tour(self, ax: plt.Axes, state: State) -> None: