diff --git a/auto_labeling_pipeline/models.py b/auto_labeling_pipeline/models.py index bb06f99..7c00cec 100644 --- a/auto_labeling_pipeline/models.py +++ b/auto_labeling_pipeline/models.py @@ -39,12 +39,16 @@ def get_all_subclasses(cls): def find_and_replace_value(obj, value, target='{{ text }}'): - for k, v in obj.items(): - if v == target: - obj[k] = value - return - if isinstance(v, dict): - find_and_replace_value(v, value, target) + if isinstance(obj, dict): + for k, v in obj.items(): + if v == target: + obj[k] = value + return + if isinstance(v, (dict, list)): + find_and_replace_value(v, value, target) + if isinstance(obj, list): + for item in obj: + find_and_replace_value(item, value, target) class CustomRESTRequestModel(RequestModel):