From 2c6a5c5b96d7166faacf077e145bd9829363cb33 Mon Sep 17 00:00:00 2001
From: NiranthS <44475481+NiranthS@users.noreply.github.com>
Date: Thu, 23 Apr 2020 23:08:29 +0530
Subject: [PATCH] Update getCameraImageTest.py

---
 examples/pybullet/examples/getCameraImageTest.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/examples/pybullet/examples/getCameraImageTest.py b/examples/pybullet/examples/getCameraImageTest.py
index c68888577..ceae99566 100644
--- a/examples/pybullet/examples/getCameraImageTest.py
+++ b/examples/pybullet/examples/getCameraImageTest.py
@@ -2,10 +2,12 @@ import matplotlib.pyplot as plt
 import numpy as np
 import pybullet as p
 import time
+import pybullet_data
 
 direct = p.connect(p.GUI)  #, options="--window_backend=2 --render_device=0")
 #egl = p.loadPlugin("eglRendererPlugin")
 
+p.setAdditionalSearchPath(pybullet_data.getDataPath())
 p.loadURDF('plane.urdf')
 p.loadURDF("r2d2.urdf", [0, 0, 1])
 p.loadURDF('cube_small.urdf', basePosition=[0.0, 0.0, 0.025])