Initial word alignment build job

sillsdev · Aug 15, 2024 · bff4239 · bff4239
1 parent 85138a5
commit bff4239
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -5,9 +5,16 @@
   },
   "python.testing.unittestEnabled": false,
   "python.testing.pytestEnabled": true,
+  "python.analysis.extraPaths": [
+    "tests"
+  ],
   "[python]": {
     "editor.defaultFormatter": "ms-python.black-formatter",
     "editor.formatOnSave": true
   },
-  "black-formatter.path": ["poetry", "run", "black"]
-}
+  "black-formatter.path": [
+    "poetry",
+    "run",
+    "black"
+  ]
+}
diff --git a/machine/jobs/__init__.py b/machine/jobs/__init__.py
@@ -2,18 +2,22 @@
 from .local_shared_file_service import LocalSharedFileService
 from .nmt_engine_build_job import NmtEngineBuildJob
 from .nmt_model_factory import NmtModelFactory
-from .shared_file_service import PretranslationInfo, PretranslationWriter, SharedFileService
+from .shared_file_service import DictToJsonWriter, PretranslationInfo, SharedFileService
 from .smt_engine_build_job import SmtEngineBuildJob
 from .smt_model_factory import SmtModelFactory
+from .word_alignment_build_job import WordAlignmentBuildJob
+from .word_alignment_model_factory import WordAlignmentModelFactory
 
 __all__ = [
     "ClearMLSharedFileService",
     "LocalSharedFileService",
     "NmtEngineBuildJob",
     "NmtModelFactory",
     "PretranslationInfo",
-    "PretranslationWriter",
+    "DictToJsonWriter",
     "SharedFileService",
     "SmtEngineBuildJob",
     "SmtModelFactory",
+    "WordAlignmentBuildJob",
+    "WordAlignmentModelFactory",
 ]