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

---
 examples/pybullet/examples/segmask_linkindex.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/examples/pybullet/examples/segmask_linkindex.py b/examples/pybullet/examples/segmask_linkindex.py
index a32fb037f..6733ce3e1 100644
--- a/examples/pybullet/examples/segmask_linkindex.py
+++ b/examples/pybullet/examples/segmask_linkindex.py
@@ -1,5 +1,8 @@
 import pybullet as p
+import pybullet_data
+
 p.connect(p.GUI)
+p.setAdditionalSearchPath(pybullet_data.getDataPath())
 r2d2 = p.loadURDF("r2d2.urdf", [0, 0, 1])
 for l in range(p.getNumJoints(r2d2)):
   print(p.getJointInfo(r2d2, l))