Cleanup

tracetools_analysis/tracetools_analysis/analysis/data_model/ros.py

@@ -145,7 +145,7 @@ class RosDataModel():
 
     def print_model(self) -> None:
         """Debug method to print every contained df."""
-        print('====================DATA MODEL====================')
+        print('====================ROS DATA MODEL====================')
         print(f'Contexts:\n{self.contexts.to_string()}')
         print()
         print(f'Nodes:\n{self.nodes.to_string()}')

tracetools_analysis/tracetools_analysis/analysis/utils.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Module for data model utility class."""
+"""Module for data model utility classes."""
 
 from datetime import datetime as dt
 from typing import Any
@@ -26,7 +26,7 @@ from .data_model.ros import RosDataModel
 
 class RosDataModelUtil():
     """
-    Data model utility class.
+    ROS data model utility class.
 
     Provides functions to get info on a ROS data model.
     """

tracetools_analysis/tracetools_analysis/process.py

@@ -31,12 +31,13 @@ def parse_args():
 
 def main():
     args = parse_args()
-
     pickle_filename = args.pickle_file
 
     start_time = time.time()
+
     events = load.load_pickle(pickle_filename)
     processor = ros2_processor.ros2_process(events)
+
     time_diff = time.time() - start_time
     print(f'processed {len(events)} events in {time_diff * 1000:.2f} ms')