From f7b368658a32879a6508731d571ca3f82e5e6c76 Mon Sep 17 00:00:00 2001
From: Christophe Bedard <fixed-term.christophe.bourquebedard@de.bosch.com>
Date: Mon, 29 Jul 2019 12:56:23 +0200
Subject: [PATCH] Cleanup

---
 .../tracetools_analysis/analysis/data_model/ros.py            | 2 +-
 tracetools_analysis/tracetools_analysis/analysis/utils.py     | 4 ++--
 tracetools_analysis/tracetools_analysis/process.py            | 3 ++-
 3 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/tracetools_analysis/tracetools_analysis/analysis/data_model/ros.py b/tracetools_analysis/tracetools_analysis/analysis/data_model/ros.py
index 1fae54d..a82ed3b 100644
--- a/tracetools_analysis/tracetools_analysis/analysis/data_model/ros.py
+++ b/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()}')
diff --git a/tracetools_analysis/tracetools_analysis/analysis/utils.py b/tracetools_analysis/tracetools_analysis/analysis/utils.py
index 8ed5348..1c3b05f 100644
--- a/tracetools_analysis/tracetools_analysis/analysis/utils.py
+++ b/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.
     """
diff --git a/tracetools_analysis/tracetools_analysis/process.py b/tracetools_analysis/tracetools_analysis/process.py
index e02470e..7553b4e 100644
--- a/tracetools_analysis/tracetools_analysis/process.py
+++ b/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')