Generate NL sentences.csv under a new nl directory. (#246)

datacommonsorg · Nov 6, 2023 · 00c9b3a · 00c9b3a
1 parent f5155a1
commit 00c9b3a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/simple/sample/output/sentences.csv → simple/sample/output/nl/sentences.csv b/simple/sample/output/sentences.csv → simple/sample/output/nl/sentences.csv
diff --git a/simple/stats/constants.py b/simple/stats/constants.py
@@ -28,6 +28,7 @@
 OBSERVATIONS_FILE_NAME_PREFIX = "observations"
 DEBUG_RESOLVE_FILE_NAME_PREFIX = "debug_resolve"
 PROCESS_DIR_NAME = "process"
+NL_DIR_NAME = "nl"
 
 PROPERTY_DESCRIPTION = "description"
 """

diff --git a/simple/stats/runner.py b/simple/stats/runner.py
@@ -39,6 +39,7 @@ def __init__(
   ) -> None:
     self.input_fh = create_file_handler(input_path)
     self.output_dir_fh = create_file_handler(output_dir)
+    self.nl_dir_fh = self.output_dir_fh.make_file(f"{constants.NL_DIR_NAME}/")
     self.process_dir_fh = self.output_dir_fh.make_file(
         f"{constants.PROCESS_DIR_NAME}/")
     self.reporter = ImportReporter(report_fh=self.process_dir_fh.make_file(
@@ -57,6 +58,7 @@ def __init__(
     self.nodes = Nodes(self.config)
 
     self.output_dir_fh.make_dirs()
+    self.nl_dir_fh.make_dirs()
     self.process_dir_fh.make_dirs()
 
   def run(self):
@@ -71,7 +73,7 @@ def run(self):
       # Generate SV sentences.
       nl.generate_sv_sentences(
           list(self.nodes.variables.values()),
-          self.output_dir_fh.make_file(constants.SENTENCES_FILE_NAME))
+          self.nl_dir_fh.make_file(constants.SENTENCES_FILE_NAME))
 
       # Report done.
       self.reporter.report_done()