diff --git a/src/unitxt/schema.py b/src/unitxt/schema.py index 57a3e9b296..cc7d856163 100644 --- a/src/unitxt/schema.py +++ b/src/unitxt/schema.py @@ -166,8 +166,9 @@ def process( task_data["metadata"]["template"] = self.artifact_to_jsonable( instance["recipe_metadata"]["template"] ) - if "criteria" in task_data and isinstance(task_data["criteria"], Artifact): - task_data["criteria"] = self.artifact_to_jsonable(task_data["criteria"]) + for k in task_data.keys(): + if k.startswith("criteri") and isinstance(task_data[k], Artifact): + task_data[k] = self.artifact_to_jsonable(task_data[k]) if constants.demos_field in instance: task_data[constants.demos_field] = [ self._get_instance_task_data(instance)