examples/seq2seq/__init__.py mutates sys.path (#7194)

examples/seq2seq/distillation.py

@@ -10,37 +10,21 @@ import torch
 from torch import nn
 from torch.nn import functional as F
 
+from finetune import SummarizationModule, TranslationModule
+from finetune import main as ft_main
+from initialization_utils import copy_layers, init_student
 from lightning_base import generic_train
 from transformers import AutoModelForSeq2SeqLM, MBartTokenizer, T5Config, T5ForConditionalGeneration
 from transformers.modeling_bart import shift_tokens_right
-
-
-try:
-    from .finetune import SummarizationModule, TranslationModule
-    from .finetune import main as ft_main
-    from .initialization_utils import copy_layers, init_student
-    from .utils import (
-        any_requires_grad,
-        assert_all_frozen,
-        calculate_bleu,
-        freeze_params,
-        label_smoothed_nll_loss,
-        pickle_load,
-        use_task_specific_params,
-    )
-except ImportError:
-    from finetune import SummarizationModule, TranslationModule
-    from finetune import main as ft_main
-    from initialization_utils import copy_layers, init_student
-    from utils import (
-        any_requires_grad,
-        assert_all_frozen,
-        calculate_bleu,
-        freeze_params,
-        label_smoothed_nll_loss,
-        pickle_load,
-        use_task_specific_params,
-    )
+from utils import (
+    any_requires_grad,
+    assert_all_frozen,
+    calculate_bleu,
+    freeze_params,
+    label_smoothed_nll_loss,
+    pickle_load,
+    use_task_specific_params,
+)
 
 
 class BartSummarizationDistiller(SummarizationModule):