Commit c6b5c137 authored by Raul Puri's avatar Raul Puri
Browse files

Merge branch 'staging_fix_task_main_imports' into 'staging'

Staging fix task main imports

See merge request ADLR/megatron-lm!62
parents f1f9fa0a 4d39aba4
Loading
Loading
Loading
Loading
+3 −2
Original line number Diff line number Diff line
@@ -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."""