only import compasses when needed

SJSURoboticsTeam · Nov 23, 2024 · c3d46b2 · c3d46b2
1 parent ceda5bc
commit c3d46b2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/compass/compass/main.py b/src/compass/compass/main.py
@@ -1,5 +1,3 @@
-from compass.actual_compass import ActualCompass
-from compass.fake_compass import FakeCompass
 from std_msgs.msg import Float64
 from constants import COMPASS_TOPIC, QOS
 from rclpy.node import Node
@@ -20,8 +18,12 @@ def __init__(self):
         )
 
         if self.get_parameter("compass_type").value == "actual":
+            from compass.actual_compass import ActualCompass
+
             self.compass = ActualCompass()
         elif self.get_parameter("compass_type").value == "fake":
+            from compass.fake_compass import FakeCompass
+
             self.compass = FakeCompass()
         else:
             raise ValueError("Unknown compass type")