"examples/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "1990e797e1f5ef5bc12dbdde6ee03c8066754bf1"
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