[lit] Make main.py a pure Python module

Running it directly as a tool, that is what lit.py is for.

1 files changed, 3 insertions, 4 deletions

diff --git a/llvm/utils/lit/lit.py b/llvm/utils/lit/lit.py
index af070c6..2f1fb43 100755
--- a/llvm/utils/lit/lit.py
+++ b/llvm/utils/lit/lit.py
@@ -1,7 +1,6 @@
 #!/usr/bin/env python
-import sys
 
-from lit.main import main
+import lit.main
 
-if __name__=='__main__':
-    main()
+if __name__ == '__main__':
+    lit.main()