Fix parse "None" as string (#33)

anand-nv · pre-commit-ci[bot] · web-flow · commit 1b81616e00ce · 2023-02-06T10:28:12.000-08:00
* Fix parse "None" as string Signed-off-by: Anand Joseph <anajoseph@nvidia.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Signed-off-by: Anand Joseph <anajoseph@nvidia.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
diff --git a/tools/text_processing_deployment/pynini_export.py b/tools/text_processing_deployment/pynini_export.py
@@ -92,7 +92,7 @@ def parse_args():
         "e.g., for English whitelist files are stored under inverse_normalization/en/data/whitelist. If None,"
         "the default file will be used.",
         default=None,
-        type=str,
+        type=lambda x: None if x == "None" else x,
     )
     parser.add_argument("--overwrite_cache", help="set to True to re-create .far grammar files", action="store_true")
     parser.add_argument(

Original file line number	Diff line number	Diff line change
`@@ -92,7 +92,7 @@ def parse_args():`
`92`	`92`	`"e.g., for English whitelist files are stored under inverse_normalization/en/data/whitelist. If None,"`
`93`	`93`	`"the default file will be used.",`
`94`	`94`	`default=None,`
`95`		`- type=str,`
	`95`	`+ type=lambda x: None if x == "None" else x,`
`96`	`96`	`)`
`97`	`97`	`parser.add_argument("--overwrite_cache", help="set to True to re-create .far grammar files", action="store_true")`
`98`	`98`	`parser.add_argument(`