From 9da572c9dac2d2ef491f95d47d17ad9e3974de90 Mon Sep 17 00:00:00 2001 From: Nate Shim Date: Wed, 8 Jan 2025 11:48:11 -0800 Subject: [PATCH] add as_json tag --- validmind/tests/load.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/validmind/tests/load.py b/validmind/tests/load.py index 1d8da6b07..a1731f27d 100644 --- a/validmind/tests/load.py +++ b/validmind/tests/load.py @@ -191,7 +191,7 @@ def list_tags(): return list(unique_tags) -def list_tasks_and_tags(): +def list_tasks_and_tags(as_json=False): """ List all task types and their associated tags, with one row per task type and all tags for a task type in one row. @@ -205,6 +205,9 @@ def list_tasks_and_tags(): for task in test.__tasks__: task_tags_dict.setdefault(task, set()).update(test.__tags__) + if as_json: + return task_tags_dict + return format_dataframe( pd.DataFrame( [