update cv2 import to work with pyrobot

examples/shortest_path_follower_example.py

@@ -7,7 +7,13 @@
 import os
 import shutil
 
-import cv2
+# TODO(akadian): remove the below pyrobot hack
+import sys
+ros_path = '/opt/ros/kinetic/lib/python2.7/dist-packages'
+if ros_path in sys.path:
+    sys.path.remove(ros_path)
+    import cv2
+sys.path.append(ros_path)
 import numpy as np
 
 import habitat

habitat/tasks/nav/nav_task.py

@@ -7,7 +7,13 @@
 from typing import Any, List, Optional, Type
 
 import attr
-import cv2
+# TODO(akadian): remove the below pyrobot hack
+import sys
+ros_path = '/opt/ros/kinetic/lib/python2.7/dist-packages'
+if ros_path in sys.path:
+    sys.path.remove(ros_path)
+    import cv2
+sys.path.append(ros_path)
 import numpy as np
 from gym import spaces
 

habitat/utils/visualizations/maps.py

@@ -7,7 +7,13 @@
 import os
 from typing import List, Optional, Tuple
 
-import cv2
+# TODO(akadian): remove the below pyrobot hack
+import sys
+ros_path = '/opt/ros/kinetic/lib/python2.7/dist-packages'
+if ros_path in sys.path:
+    sys.path.remove(ros_path)
+    import cv2
+sys.path.append(ros_path)
 import imageio
 import numpy as np
 import scipy.ndimage

habitat/utils/visualizations/utils.py

@@ -7,7 +7,13 @@
 import os
 from typing import Dict, List, Optional, Tuple
 
-import cv2
+# TODO(akadian): remove the below pyrobot hack
+import sys
+ros_path = '/opt/ros/kinetic/lib/python2.7/dist-packages'
+if ros_path in sys.path:
+    sys.path.remove(ros_path)
+    import cv2
+sys.path.append(ros_path)
 import imageio
 import numpy as np
 import tqdm