update image path

sdpython · Dec 22, 2017 · 8ae6a31 · 8ae6a31
1 parent 167de32
commit 8ae6a31
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/_doc/examples/plot_rest_api_search_images.py b/_doc/examples/plot_rest_api_search_images.py
@@ -160,6 +160,9 @@ def path_module(mod):
     txts = list(map(lambda x: str(x[0]), res))
     imgs = list(map(lambda x: os.path.join(
         'temp_scripts', 'images', x[1]), res))
+    if not os.path.exists(imgs[0]):
+        imgs = list(map(lambda x: os.path.join(
+            'images', x[1]), res))
 
     from mlinsights.plotting import plot_gallery_images
     plot_gallery_images(imgs, txts)