Print duration of callback app (#885)

Signed-off-by: Victor Chang <[email protected]>
Project-MONAI · Oct 5, 2023 · a4c63ac · a4c63ac
1 parent 387803f
commit a4c63ac
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/TaskManager/CallbackApp/app.py b/src/TaskManager/CallbackApp/app.py
@@ -15,6 +15,7 @@
 import pika
 import uuid
 
+from time import perf_counter as pc
 
 def main():
     parser = argparse.ArgumentParser()
@@ -50,4 +51,9 @@ def main():
     print('Message sent.')
 
 if __name__ == "__main__":
-    main()
+    t0 = pc()
+    try:
+        main()
+    except Exception as e:
+        print(e)
+    print(f"Duration: {pc()-t0}")