fix dataset name to be robot name

BerkeleyAutomation · Apr 29, 2024 · c49b7fa · c49b7fa
1 parent c2e64d3
commit c49b7fa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tutorial_workspace/fogros2_tutorial/fog_rtx_recorder.py b/tutorial_workspace/fogros2_tutorial/fog_rtx_recorder.py
@@ -11,9 +11,9 @@ class CompressedImageSubscriber(Node):
     def __init__(self):
         super().__init__('compressed_image_subscriber')
 
-        self.declare_parameter('dataset_name', 'demo_ds')
-        self.dataset_name = self.get_parameter('dataset_name').value
-
+        self.dataset_name = self.declare_parameter('robot_name', 'PARAMETER NOT SET').get_parameter_value().string_value
+        if self.dataset_name == 'PARAMETER NOT SET':
+            raise ValueError('Please set the dataset_name parameter')
         self.declare_parameter("max_idle_time", 10)
         self.max_idle_time = self.get_parameter("max_idle_time").value
 

diff --git a/tutorial_workspace/launch/cloudgripper.launch.py b/tutorial_workspace/launch/cloudgripper.launch.py
@@ -37,7 +37,7 @@ def generate_launch_description():
         Node(
             package="fogros2_tutorial", 
             executable="fog_rtx_recorder", 
-            parameters=[{'dataset_name': LaunchConfiguration('robot_name')}]
+            parameters=[{'robot_name': LaunchConfiguration('robot_name')}]
         ), 
         Node(
             package="fogros2_tutorial", executable="sam_client",