Make main into a function so we can profile it

1 files changed, 6 insertions, 1 deletions

diff --git a/bin/chroma-sim b/bin/chroma-sim
index 7f309de..5454ce3 100755
--- a/bin/chroma-sim
+++ b/bin/chroma-sim
@@ -1,7 +1,9 @@
 #!/usr/bin/env python
 #--*-python-*-
+from chroma.tools import profile_if_possible
 
-if __name__ == '__main__':
+@profile_if_possible
+def main():
     import optparse
     import sys
     import imp
@@ -109,3 +111,6 @@ if __name__ == '__main__':
 
     writer.close()
 
+if __name__ == '__main__':
+    main()
+