fix: handle text as categoricals if sensitive=True (#1415)

ydataai · Aug 7, 2023 · 346478c · 346478c
1 parent 53dd27c
commit 346478c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/ydata_profiling/report/structure/variables/render_categorical.py b/src/ydata_profiling/report/structure/variables/render_categorical.py
@@ -339,10 +339,14 @@ def render_categorical(config: Settings, summary: dict) -> dict:
 
     template_variables = render_common(config, summary)
 
+    type_name = summary["type"]
+    if isinstance(type_name, list):
+        type_name = type_name[0]
+
     info = VariableInfo(
         summary["varid"],
         summary["varname"],
-        "Categorical",
+        type_name,
         summary["alerts"],
         summary["description"],
         style=config.html.style,

diff --git a/src/ydata_profiling/report/structure/variables/render_text.py b/src/ydata_profiling/report/structure/variables/render_text.py
@@ -12,6 +12,7 @@
 from ydata_profiling.report.structure.variables.render_categorical import (
     _get_n,
     freq_table,
+    render_categorical,
     render_categorical_frequency,
     render_categorical_length,
     render_categorical_unicode,
@@ -21,6 +22,10 @@
 
 
 def render_text(config: Settings, summary: Dict[str, Any]) -> Dict[str, Any]:
+    if config.vars.text.redact:
+        render = render_categorical(config, summary)
+        return render
+
     varid = summary["varid"]
     words = config.vars.text.words
     characters = config.vars.text.characters