Merge branch 'staging_fix_task_main_imports' into 'staging'

Staging fix task main imports

See merge request ADLR/megatron-lm!62

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."""