ref: add zwo as alternative for 2

souvikg10 · Oct 2, 2023 · eaac7ea · eaac7ea
1 parent e0f77a4
commit eaac7ea
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 2 deletions.
diff --git a/Duckling/Numeral/DE/Corpus.hs b/Duckling/Numeral/DE/Corpus.hs
@@ -36,6 +36,11 @@ allExamples = concat
              , "Eine"
              , "einen"
              ]
+  , examples (NumeralValue 2)
+             [ "2"
+             , "Zwei"
+             , "Zwo"
+             ]
   , examples (NumeralValue 3)
              [ "3"
              , "Drei"
@@ -77,6 +82,16 @@ allExamples = concat
              [ "18"
              , "achtzehn"
              ]
+  , examples (NumeralValue 82)
+             [ "82"
+             , "zwoundachtzig"
+             , "zweiundachtzig"
+             ]
+  , examples (NumeralValue 182)
+             [ "182"
+             , "einhundertzwoundachtzig"
+             , "einhundertzweiundachtzig"
+             ]
   , examples (NumeralValue 200)
              [ "200"
              , "zweihundert"

diff --git a/Duckling/Numeral/DE/NumParser.hs b/Duckling/Numeral/DE/NumParser.hs
@@ -103,6 +103,9 @@ two =
     , times10 = [assign 20 "zwanzig"]
     }
 
+two_alternative :: NumItem
+two_alternative = defaultNumItem 2 "zwo"
+
 three :: NumItem
 three =
   (defaultNumItem 3 "drei")
@@ -139,7 +142,7 @@ nine :: NumItem
 nine = defaultNumItem 9 "neun"
 
 digitLexicon :: [NumItem]
-digitLexicon = [one, two, three, four, five, six, seven, eight, nine]
+digitLexicon = [one, two_alternative, two, three, four, five, six, seven, eight, nine]
 
 from1to9 :: NumParser
 from1to9 = foldr ((<|>) . base) empty digitLexicon

diff --git a/Duckling/Numeral/DE/Rules.hs b/Duckling/Numeral/DE/Rules.hs
@@ -218,7 +218,7 @@ ruleIntegerWithThousandsSeparator = Rule
 ruleAllNumeralWords :: Rule
 ruleAllNumeralWords = Rule
   { name = "simple and complex numerals written as one word"
-  , pattern = [regex "(ein|zwei|drei|vier|fünf|sech|sieb|acht|neun|zehn|elf|zwölf|hundert|tausend)?([^\\s]+)?(eine[m|n|r|s]?|eins?|zwei|drei|vier|fünf|sechs|sieben|acht|neun|zehn|elf|zwölf|[s|ß|z]ig|hundert|tausend)"]
+  , pattern = [regex "(ein|zwei|zwo|drei|vier|fünf|sech|sieb|acht|neun|zehn|elf|zwölf|hundert|tausend)?([^\\s]+)?(eine[m|n|r|s]?|eins?|zwei|drei|vier|fünf|sechs|sieben|acht|neun|zehn|elf|zwölf|[s|ß|z]ig|hundert|tausend)"]
   , prod = \tokens -> case tokens of
       (Token RegexMatch (GroupMatch matches) : _) ->
         (parseNumeral $ concat $ Text.unpack . Text.toLower <$> matches)