ikarasz commited on
Commit
ea942a7
·
1 Parent(s): 20c656b

exclude some words from singular transformation

Browse files
Files changed (1) hide show
  1. utils.py +14 -0
utils.py CHANGED
@@ -223,6 +223,9 @@ MATH_WORDS = [
223
  "distribution",
224
  "distributive",
225
  "divide",
 
 
 
226
  "dividend",
227
  "divisibility test",
228
  "divisible by",
@@ -442,7 +445,12 @@ MATH_WORDS = [
442
  "modal",
443
  "mode",
444
  "multipl",
 
 
 
445
  "multiple",
 
 
446
  "multiplication counting principle",
447
  "multiplication diagram",
448
  "multiplication fact",
@@ -805,6 +813,10 @@ MATH_WORDS = [
805
  "zero",
806
  ]
807
 
 
 
 
 
808
 
809
  p = inflect.engine()
810
 
@@ -815,6 +827,8 @@ def singular_to_plural(word):
815
 
816
  def plural_to_singular(word):
817
  """Convert plural word to singular using inflect."""
 
 
818
  return p.singular_noun(word) or word
819
 
820
  plural_MATH_WORDS = [singular_to_plural(word) for word in MATH_WORDS]
 
223
  "distribution",
224
  "distributive",
225
  "divide",
226
+ "divided",
227
+ "divides",
228
+ "dividing",
229
  "dividend",
230
  "divisibility test",
231
  "divisible by",
 
445
  "modal",
446
  "mode",
447
  "multipl",
448
+ "multiply",
449
+ "multiplied",
450
+ "multiplies",
451
  "multiple",
452
+ "multiplication",
453
+ "multiplying",
454
  "multiplication counting principle",
455
  "multiplication diagram",
456
  "multiplication fact",
 
813
  "zero",
814
  ]
815
 
816
+ PLURAL_TO_SINGULAR_EXCLUSIONS = [
817
+ "plus",
818
+ "minus",
819
+ ]
820
 
821
  p = inflect.engine()
822
 
 
827
 
828
  def plural_to_singular(word):
829
  """Convert plural word to singular using inflect."""
830
+ if word in PLURAL_TO_SINGULAR_EXCLUSIONS:
831
+ return word
832
  return p.singular_noun(word) or word
833
 
834
  plural_MATH_WORDS = [singular_to_plural(word) for word in MATH_WORDS]