enabled passing one alignment at a time to map

unfoldingWord · Nov 21, 2018 · c974401 · c974401
1 parent 2fbec1f
commit c974401
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "wordmap",
-  "version": "0.4.2",
+  "version": "0.4.3",
   "description": "Multi-Lingual Word Alignment Prediction",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

diff --git a/src/WordMap.ts b/src/WordMap.ts
@@ -83,8 +83,16 @@ export default class WordMap {
     this.engine.addCorpus(sourceTokens, targetTokens);
   }
 
-  public appendAlignmentMemory(alignments: Alignment[]) {
-    this.engine.addAlignmentMemory(alignments);
+  /**
+   * Appends alignment memory engine.
+   * @param alignments - an alignment or array of alignments
+   */
+  public appendAlignmentMemory(alignments: Alignment | Alignment[]) {
+    if (alignments instanceof Array) {
+      this.engine.addAlignmentMemory(alignments);
+    } else {
+      this.engine.addAlignmentMemory([alignments]);
+    }
   }
 
   /**