Format with black

Author: GiulioRomualdi

robot_log_visualizer/file_reader/signal_provider.py

@@ -96,14 +96,16 @@ def __populate_text_logging_data(self, file_object):
                     # If len(value[text[0]].shape) == 2 then the text contains a string, otherwise it is empty
                     # We need to manually check the shape to handle the case in which the text is empty
                     data[key]["data"] = [
-                        TextLoggingMsg(
-                            text="".join(chr(c[0]) for c in value[text[0]]),
-                            level="".join(chr(c[0]) for c in value[level[0]]),
-                        )
-                        if len(value[text[0]].shape) == 2
-                        else TextLoggingMsg(
-                            text="",
-                            level="".join(chr(c[0]) for c in value[level[0]]),
+                        (
+                            TextLoggingMsg(
+                                text="".join(chr(c[0]) for c in value[text[0]]),
+                                level="".join(chr(c[0]) for c in value[level[0]]),
+                            )
+                            if len(value[text[0]].shape) == 2
+                            else TextLoggingMsg(
+                                text="",
+                                level="".join(chr(c[0]) for c in value[level[0]]),
+                            )
                         )
                         for text, level in zip(text_ref, level_ref)
                     ]
@@ -217,7 +219,7 @@ def robot_state_path(self):
         locker = QMutexLocker(self.robot_state_path_lock)
         value = self._robot_state_path
         return value
-    
+
     @property
     def max_arrow(self):
         locker = QMutexLocker(self._max_arrow_mutex)
@@ -399,7 +401,9 @@ def register_3d_arrow(self, key, arrow_path):
             data, _ = self.get_item_from_path(arrow_path)
             arrow = data[:, 3:]
             self._max_arrow_mutex.lock()
-            self._max_arrow = max(np.max(np.linalg.norm(arrow, axis=1)), self._max_arrow)
+            self._max_arrow = max(
+                np.max(np.linalg.norm(arrow, axis=1)), self._max_arrow
+            )
             self._max_arrow_mutex.unlock()
         self._3d_arrows_path_lock.unlock()
 

robot_log_visualizer/ui/gui.py

@@ -103,7 +103,7 @@ def get_urdf_path(self):
 
     def get_package_directory(self):
         return self.ui.packageDirLineEdit.text()
-    
+
     def buttonBox_on_click(self, button):
         self.clicked_button = button
 
@@ -113,12 +113,12 @@ def get_clicked_button_role(self):
         if self.clicked_button is not None:
             return self.ui.buttonBox.buttonRole(self.clicked_button)
         return None
-    
+
     def get_clicked_button_text(self):
         if self.clicked_button is not None:
             return self.clicked_button.text()
         return None
-    
+
     def get_clicked_standard_button(self):
         return self.std_button
 
@@ -131,6 +131,7 @@ def handle_arrow_scaling(self):
             self.ui.arrowScaling_lineEdit.setText("")
             self.ui.arrowScaling_lineEdit.setEnabled(True)
 
+
 class About(QtWidgets.QMainWindow):
     def __init__(self):
         # call QMainWindow constructor
@@ -727,24 +728,26 @@ def open_set_robot_model(self):
             if std_button == QtWidgets.QDialogButtonBox.SaveAll:
                 if not self.dataset_loaded:
                     self.meshcat_provider.model_path = dlg.get_urdf_path()
-                    self.meshcat_provider.custom_package_dir = dlg.get_package_directory()
-                
+                    self.meshcat_provider.custom_package_dir = (
+                        dlg.get_package_directory()
+                    )
 
                 arrow_scaling_value = dlg.ui.arrowScaling_lineEdit.text()
                 if not arrow_scaling_value:
                     arrow_scaling_value = "1.0"
                 else:
                     arrow_scaling_value = float(arrow_scaling_value)
                 self.signal_provider.set_custom_max_arrow(
-                    not dlg.ui.arrowScaling_checkBox.isChecked(),
-                    arrow_scaling_value
+                    not dlg.ui.arrowScaling_checkBox.isChecked(), arrow_scaling_value
                 )
             if std_button == QtWidgets.QDialogButtonBox.Save:
                 # we need to check which tab is selected in the dlg
                 if dlg.ui.tabWidget.currentIndex() == 0:
                     if not self.dataset_loaded:
                         self.meshcat_provider.model_path = dlg.get_urdf_path()
-                        self.meshcat_provider.custom_package_dir = dlg.get_package_directory()
+                        self.meshcat_provider.custom_package_dir = (
+                            dlg.get_package_directory()
+                        )
                 else:
                     arrow_scaling_value = dlg.ui.arrowScaling_lineEdit.text()
                     # if it is empty we set it to 1.0
@@ -754,7 +757,7 @@ def open_set_robot_model(self):
                         arrow_scaling_value = float(arrow_scaling_value)
                     self.signal_provider.set_custom_max_arrow(
                         not dlg.ui.arrowScaling_checkBox.isChecked(),
-                        arrow_scaling_value
+                        arrow_scaling_value,
                     )
 
             else: