fixed code

controller_learning/iterative_learning_trajectories.py

@@ -6,7 +6,7 @@
 import time
 import numpy as np
 import rospy
-from ControllerLearning import TrajectoryLearning
+from src.ControllerLearning import TrajectoryLearning
 from std_msgs.msg import Bool
 from common import update_mpc_params, setup_sim, random_replace, initialize_vio
 

controller_learning/src/ControllerLearning/TrajectoryBase.py

@@ -9,6 +9,9 @@
 import numpy as np
 import rospy
 from nav_msgs.msg import Odometry
+# from quadrotor_msgs.msg import ControlCommand
+import sys
+sys.path.insert(0,'/home/archit/drone_acrobatics_ws/catkin_dda/devel/lib/python3/dist-packages')
 from quadrotor_msgs.msg import ControlCommand
 from quadrotor_msgs.msg import TrajectoryPoint
 from sensor_msgs.msg import PointCloud

controller_learning/src/ControllerLearning/TrajectoryLearning.py

@@ -7,14 +7,20 @@
 import numpy as np
 import rospy
 from nav_msgs.msg import Odometry
-from quadrotor_msgs.msg import ControlCommand
+
 from std_msgs.msg import Bool
 from std_msgs.msg import Empty
 
+import sys
+sys.path.insert(1,'/home/archit/drone_acrobatics_ws/catkin_dda/devel/lib/python3/dist-packages')
+from quadrotor_msgs.msg import ControlCommand
+
 from .TrajectoryBase import TrajectoryBase, TRACK_NUM_NORMALIZE
 
 
 
+
+
 class TrajectoryLearning(TrajectoryBase):
     def __init__(self, config, mode):
         super(TrajectoryLearning, self).__init__(config, mode)