From d397313626e38c58afceb6a6020a8ba2500b7075 Mon Sep 17 00:00:00 2001 From: Leon Derczynski Date: Wed, 3 Jul 2024 10:30:16 +0200 Subject: [PATCH] enable setting nested dicts directly in plugin config --- garak/_config.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/garak/_config.py b/garak/_config.py index 982fcc32..1d66e19c 100644 --- a/garak/_config.py +++ b/garak/_config.py @@ -7,6 +7,7 @@ # logging should be set up before config is loaded +from collections import defaultdict from dataclasses import dataclass import importlib import logging @@ -60,11 +61,13 @@ class TransientConfig(GarakSubConfig): run = GarakSubConfig() plugins = GarakSubConfig() reporting = GarakSubConfig() -plugins.probes = {} -plugins.generators = {} -plugins.detectors = {} -plugins.buffs = {} -plugins.harnesses = {} + +nested_dict = lambda: defaultdict(nested_dict) +plugins.probes = nested_dict() +plugins.generators = nested_dict() +plugins.detectors = nested_dict() +plugins.buffs = nested_dict() +plugins.harnesses = nested_dict() reporting.taxonomy = None # set here to enable report_digest to be called directly buffmanager = BuffManager()