Ros2 fixes (#95)

* Get rid of tests_require build warning

* Get rid of KeyboardInterrupt

* Lint

* Implement for other nodes

Author: redvinaa

yolo_ros/setup.py

@@ -16,7 +16,7 @@
     maintainer_email="mgons@unileon.es",
     description="YOLO for ROS 2",
     license="GPL-3",
-    tests_require=["pytest"],
+    extras_require={"test": ["pytest"]},
     entry_points={
         "console_scripts": [
             "yolo_node = yolo_ros.yolo_node:main",

yolo_ros/yolo_ros/debug_node.py

@@ -380,6 +380,8 @@ def main():
     node = DebugNode()
     node.trigger_configure()
     node.trigger_activate()
-    rclpy.spin(node)
-    node.destroy_node()
-    rclpy.shutdown()
+
+    try:
+        rclpy.spin(node)
+    except KeyboardInterrupt:
+        pass

yolo_ros/yolo_ros/detect_3d_node.py

@@ -441,6 +441,8 @@ def main():
     node = Detect3DNode()
     node.trigger_configure()
     node.trigger_activate()
-    rclpy.spin(node)
-    node.destroy_node()
-    rclpy.shutdown()
+
+    try:
+        rclpy.spin(node)
+    except KeyboardInterrupt:
+        pass

yolo_ros/yolo_ros/tracking_node.py

@@ -203,6 +203,8 @@ def main():
     node = TrackingNode()
     node.trigger_configure()
     node.trigger_activate()
-    rclpy.spin(node)
-    node.destroy_node()
-    rclpy.shutdown()
+
+    try:
+        rclpy.spin(node)
+    except KeyboardInterrupt:
+        pass

yolo_ros/yolo_ros/yolo_node.py

@@ -403,6 +403,8 @@ def main():
     node = YoloNode()
     node.trigger_configure()
     node.trigger_activate()
-    rclpy.spin(node)
-    node.destroy_node()
-    rclpy.shutdown()
+
+    try:
+        rclpy.spin(node)
+    except KeyboardInterrupt:
+        pass