def CategorySelector()

in community-efforts/image_preferences/01_synthetic_data_generation_prompts.py [0:0]

• 13 lines of code
• 2 McCabe index (conditional complexity)

    def CategorySelector(inputs: StepInput) -> "StepOutput":
        result = []
        for input in inputs:
            # Randomly select a category
            category = random.choice(list(categories.keys()))
            # Randomly select a subcategory from the chosen category
            subcategory = random.choice(categories[category])

            result.append(
                {
                    "category": category,
                    "subcategory": subcategory,
                    "prompt": input["prompt"],
                }
            )
        yield result