fix aligner

dwmitelman · Apr 5, 2024 · 51bd7aa · 51bd7aa
1 parent c28eb92
commit 51bd7aa
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/global_def/requirements.txt b/global_def/requirements.txt
@@ -2,3 +2,4 @@ transformers>=4.36.0
 python-docx==0.8.11
 pyarabic==0.6.15
 google-api-python-client
+tabulate
diff --git a/pre_train/aligner/align.py b/pre_train/aligner/align.py
@@ -2,9 +2,9 @@
 from tabulate import tabulate
 from typing import List
 
-from ja_transliteration_tool.pre_train.generic.const import *
-from ja_transliteration_tool.pre_train.generic.word_clean import Ar, Ja
-from ja_transliteration_tool.pre_train.aligner.transliterate import Ja2Ar, Ar2Ja
+from pre_train.generic.const import *
+from pre_train.generic.word_clean import Ar, Ja
+from pre_train.aligner.transliterate import Ja2Ar, Ar2Ja
 
 
 class Comparator(object):

diff --git a/pre_train/aligner/sample.m.py b/pre_train/aligner/sample.m.py
@@ -8,15 +8,14 @@ def sample(text_ar, text_ja):
     # print(len(split_ja), len(split_ar))
     # assert len(split_ja) == len(split_ar)
 
-    Aligner(split_ar, split_ja)._run()
+    return Aligner(split_ar, split_ja)
 
     # for i in range(len(split_ar)):
     #     assert Comparator(split_ar[i], split_ja[i]).compare()
 
 
-# text_ar = "سُئِلْتُ عمّا عنديَ من الاحتجاج خخخخخخخ على مُخالفينا من الفلاسفة وأهل الأديان ثمّ على الخوارج الذين يخالفون الجمهور"
-# text_ja = "סילת עמא ענדי מן אלאחתג'אג' עלי מכ'אלפינא מן <אלפלאספה> ואה<ל אלאדיאן> ת'ם עלי אלכ'וארג עעעעעעעע אלד'ין יכ'אלפון אלג'מהור"
-#
-# sample(text_ar, text_ja)
-
+text_ar = "سُئِلْتُ عمّا عنديَ من الاحتجاج خخخخخخخ على مُخالفينا من الفلاسفة وأهل الأديان ثمّ على الخوارج الذين يخالفون الجمهور"
+text_ja = "סילת עמא ענדי מן אלאחתג'אג' עלי מכ'אלפינא מן <אלפלאספה> ואה<ל אלאדיאן> ת'ם עלי אלכ'וארג עעעעעעעע אלד'ין יכ'אלפון אלג'מהור"
 
+res = sample(text_ar, text_ja).get_tws()
+print(res)