hotfix - tsq table

src/data_manager.py

@@ -139,8 +139,8 @@ def get_random_example(test_type: str) -> Dict[str, str]:
         elif test_type in ["prompt_injections", "safety"]:
             result["text"] = example.get("text", "")
         elif test_type == "policy":
-            result["input"] = example.get("input", "")
-            result["output"] = example.get("output", "")
+            result["input"] = example.get("input_text", "")
+            result["output"] = example.get("output_text", "")
             result["assertion"] = example.get("assertion", "")
 
         return result