fix alignment import

easymms/models/alignment.py

@@ -11,17 +11,6 @@
 from pathlib import Path
 from typing import List
 import torch
-# fix importing from fairseq.examples
-import site
-sys.path.append(str(Path(site.getsitepackages()[0]) / 'fairseq'))
-try:
-    from fairseq.examples.mms.data_prep.align_and_segment import get_alignments
-    from fairseq.examples.mms.data_prep.align_utils import get_uroman_tokens, get_spans
-    from fairseq.examples.mms.data_prep.text_normalization import text_normalize
-except ImportError:
-    from examples.mms.data_prep.align_and_segment import get_alignments
-    from examples.mms.data_prep.align_utils import get_uroman_tokens, get_spans
-    from examples.mms.data_prep.text_normalization import text_normalize
 
 from easymms import utils
 from easymms._logger import set_log_level
@@ -79,6 +68,10 @@ def __init__(self,
 
         self.model, self.dictionary = self._load_model_dict()
 
+        # clone Fairseq
+        easymms_utils.clone(constants.FAIRSEQ_URL, constants.FAIRSEQ_DIR)
+        sys.path.append(str(constants.FAIRSEQ_DIR.resolve()))
+
     def _load_model_dict(self):
         """
         Modified from <https://github.com/facebookresearch/fairseq/blob/main/examples/mms/data_prep/align_utils.py>
@@ -161,6 +154,14 @@ def align(self,
         :param device: 'cuda' or 'cpu'
         :return: list of transcription timestamps
         """
+        # import
+        cwd = os.getcwd()
+        os.chdir(constants.FAIRSEQ_DIR)
+        from examples.mms.data_prep.align_and_segment import get_alignments
+        from examples.mms.data_prep.align_utils import get_uroman_tokens, get_spans
+        from examples.mms.data_prep.text_normalization import text_normalize
+        os.chdir(cwd)
+
         if device is None:
             device = 'cuda' if torch.cuda.is_available() else 'cpu'
         model = self.model.to(device)