diff --git a/examples/shortest_path_follower_example.py b/examples/shortest_path_follower_example.py index b70c69db98eb2d66d88b9993c3a78d2721ea4deb..d62d7a643b73afd17ac31751aaaae4005ac8ca40 100644 --- a/examples/shortest_path_follower_example.py +++ b/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 diff --git a/habitat/tasks/nav/nav_task.py b/habitat/tasks/nav/nav_task.py index 3aac5d9aa8965eeea9d1bf623b2966dcf096e2c7..0c67b8c7a15e4fa99f34d74206ef09a8c7ee3de2 100644 --- a/habitat/tasks/nav/nav_task.py +++ b/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 diff --git a/habitat/utils/visualizations/maps.py b/habitat/utils/visualizations/maps.py index 1028bf3802bf9eed9da62bd8b69f640a0327d735..a46f4e695698827f71a0e200cad2dbab8e386acc 100644 --- a/habitat/utils/visualizations/maps.py +++ b/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 diff --git a/habitat/utils/visualizations/utils.py b/habitat/utils/visualizations/utils.py index 48ffc02ba396c366686bc56ae30ad288c38073d2..687558a1f0584bc35f10e2e04bef3052b8fcaf82 100644 --- a/habitat/utils/visualizations/utils.py +++ b/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