From 9f1747f999965cc83d2f93139c01816dfd324461 Mon Sep 17 00:00:00 2001
From: Patrick von Platen <patrick.v.platen@gmail.com>
Date: Tue, 3 Nov 2020 13:56:41 +0100
Subject: [PATCH] [Seq2Seq] Correct import in Seq2Seq Trainer (#8254)

---
 examples/seq2seq/seq2seq_trainer.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/examples/seq2seq/seq2seq_trainer.py b/examples/seq2seq/seq2seq_trainer.py
index 8ad1d7a6ee..dc4aeca593 100644
--- a/examples/seq2seq/seq2seq_trainer.py
+++ b/examples/seq2seq/seq2seq_trainer.py
@@ -62,10 +62,7 @@ class Seq2SeqTrainer(Trainer):
             self.loss_fn = torch.nn.CrossEntropyLoss(ignore_index=self.config.pad_token_id)
         else:
             # dynamically import label_smoothed_nll_loss
-            try:
-                from .utils import label_smoothed_nll_loss
-            except ImportError:
-                from utils import label_smoothed_nll_loss
+            from utils import label_smoothed_nll_loss
 
             self.loss_fn = label_smoothed_nll_loss