cannot run set origin.py script

[Illusion-1404]

I am getting this strange error , even though yaml is installed for python 2.7

Traceback (most recent call last):
File "/home/octo/catkin_ws/src/vision_to_mavros/scripts/set_origin.py", line 9, in
import rospy
File "/opt/ros/melodic/lib/python2.7/dist-packages/rospy/init.py", line 47, in
from std_msgs.msg import Header
File "/opt/ros/melodic/lib/python2.7/dist-packages/std_msgs/msg/init.py", line 1, in
from ._Bool import *
File "/opt/ros/melodic/lib/python2.7/dist-packages/std_msgs/msg/_Bool.py", line 6, in
import genpy
File "/opt/ros/melodic/lib/python2.7/dist-packages/genpy/init.py", line 34, in
from . message import Message, SerializationError, DeserializationError, MessageException, struct_I
File "/opt/ros/melodic/lib/python2.7/dist-packages/genpy/message.py", line 48, in
import yaml
ModuleNotFoundError: No module named 'yaml'

[thien94]

You could try:

• verify which python version is the default one. If possible, install yaml in python3 as well,
• run the script in python2 and python3 and see which is the correct version.