From b15297ae4096c9f8f0e745a8dd3b4a4505ae63db Mon Sep 17 00:00:00 2001
From: Varun Agrawal <varagrawal@gmail.com>
Date: Thu, 28 Oct 2021 15:19:36 -0400
Subject: [PATCH] address review comments

---
 python/gtsam/examples/ImuFactorExample.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/python/gtsam/examples/ImuFactorExample.py b/python/gtsam/examples/ImuFactorExample.py
index 13ed24c6c..c86a4e216 100644
--- a/python/gtsam/examples/ImuFactorExample.py
+++ b/python/gtsam/examples/ImuFactorExample.py
@@ -137,7 +137,14 @@ class ImuFactorExample(PreintegrationExample):
             T: int = 12,
             compute_covariances: bool = False,
             verbose: bool = True):
-        """Main runner."""
+        """
+        Main runner.
+
+        Args:
+            T: Total trajectory time.
+            compute_covariances: Flag indicating whether to compute marginal covariances.
+            verbose: Flag indicating if printing should be verbose.
+        """
         graph = gtsam.NonlinearFactorGraph()
 
         # initialize data structure for pre-integrated IMU measurements