fixed tasks/main imports

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