Commit ab2fa185 authored by Myle Ott's avatar Myle Ott Committed by Facebook Github Bot
Browse files

Move task import in MultilingualTransformer to fix circular dependencies

Summary: Pull Request resolved: https://github.com/fairinternal/fairseq-py/pull/687

Differential Revision: D16005399

Pulled By: myleott

fbshipit-source-id: bf099c17e2095394acc452e9abcb4ee04afd0426
parent b3864b28
...@@ -20,7 +20,6 @@ from fairseq.models.transformer import ( ...@@ -20,7 +20,6 @@ from fairseq.models.transformer import (
TransformerEncoder, TransformerEncoder,
TransformerDecoder, TransformerDecoder,
) )
from fairseq.tasks.multilingual_translation import MultilingualTranslationTask
@register_model('multilingual_transformer') @register_model('multilingual_transformer')
...@@ -59,6 +58,7 @@ class MultilingualTransformerModel(FairseqMultiModel): ...@@ -59,6 +58,7 @@ class MultilingualTransformerModel(FairseqMultiModel):
@classmethod @classmethod
def build_model(cls, args, task): def build_model(cls, args, task):
"""Build a new model instance.""" """Build a new model instance."""
from fairseq.tasks.multilingual_translation import MultilingualTranslationTask
assert isinstance(task, MultilingualTranslationTask) assert isinstance(task, MultilingualTranslationTask)
# make sure all arguments are present in older models # make sure all arguments are present in older models
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment