Add significance model generator cli

2 files changed, 7 insertions, 4 deletions

diff --git a/linguistics/src/main/java/com/yahoo/language/significance/impl/DocumentFrequencyFile.java b/linguistics/src/main/java/com/yahoo/language/significance/impl/DocumentFrequencyFile.java
index b62754ac8ad..9b7cbae834a 100644
--- a/linguistics/src/main/java/com/yahoo/language/significance/impl/DocumentFrequencyFile.java
+++ b/linguistics/src/main/java/com/yahoo/language/significance/impl/DocumentFrequencyFile.java
@@ -17,15 +17,14 @@ import java.util.HashMap;
 public class DocumentFrequencyFile {
     private final String description;
 
-    private final int documentCount;
-
+    private final long documentCount;
 
     private final HashMap<String, Long> frequencies;
 
     @JsonCreator
     public DocumentFrequencyFile(
             @JsonProperty("description") String description,
-            @JsonProperty("document-count") int documentCount,
+            @JsonProperty("document-count") long documentCount,
             @JsonProperty("document-frequencies") HashMap<String, Long> frequencies) {
         this.description = description;
         this.documentCount = documentCount;
@@ -36,7 +35,7 @@ public class DocumentFrequencyFile {
     public String description() { return description; }
 
     @JsonProperty("document-count")
-    public int documentCount() { return documentCount; }
+    public long documentCount() { return documentCount; }
 
     @JsonProperty("document-frequencies")
     public HashMap<String, Long> frequencies() { return frequencies; }
diff --git a/linguistics/src/main/java/com/yahoo/language/significance/impl/SignificanceModelFile.java b/linguistics/src/main/java/com/yahoo/language/significance/impl/SignificanceModelFile.java
index 902613379f0..94030108671 100644
--- a/linguistics/src/main/java/com/yahoo/language/significance/impl/SignificanceModelFile.java
+++ b/linguistics/src/main/java/com/yahoo/language/significance/impl/SignificanceModelFile.java
@@ -45,4 +45,8 @@ public class SignificanceModelFile {
 
     @JsonProperty("languages")
     public HashMap<String, DocumentFrequencyFile> languages() { return languages; }
+
+    public void addLanguage(String language, DocumentFrequencyFile documentFrequencyFile) {
+        languages.put(language, documentFrequencyFile);
+    }
 }