Add column info to report

opendp · Dec 3, 2024 · 2ac3184 · 2ac3184
1 parent 2344885
commit 2ac3184
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/dp_wizard/utils/code_generators/__init__.py b/dp_wizard/utils/code_generators/__init__.py
@@ -144,6 +144,7 @@ def _make_extra_blocks(self):
             Template("reports")
             .fill_expressions(
                 OUTPUTS=outputs_expression,
+                COLUMNS={k: v._asdict() for k, v in self.columns.items()},
             )
             .fill_values(
                 CSV_PATH=self.csv_path,

diff --git a/dp_wizard/utils/code_generators/no-tests/_reports.py b/dp_wizard/utils/code_generators/no-tests/_reports.py
@@ -38,6 +38,7 @@ def flatten_dict(dictionary, parent_key=""):
     "inputs": {
         "data": CSV_PATH,
         "epsilon": EPSILON,
+        "columns": COLUMNS,
     },
     "outputs": OUTPUTS,
 }