fixed tasks/main imports

tasks/main.py

@@ -15,13 +15,14 @@
 
 """Main tasks functionality."""
 
-from megatron.initialize import initialize_megatron
-from megatron import get_args
 import os
 import sys
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__),
                                              os.path.pardir)))
 
+from megatron import get_args
+from megatron.initialize import initialize_megatron
+
 
 def get_tasks_args(parser):
     """Provide extra arguments required for tasks."""