def get_query_runner()

in identity-resolution/notebooks/identity-graph/nepytune/benchmarks/query_runner.py [0:0]

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

def get_query_runner(query, samples):
    """Query runner factory."""
    if query == 'get_sibling_attrs':
        return SiblingsAttrsRunner(samples)
    elif query == 'brand_interaction_audience':
        return BrandInteractionRunner(samples)
    elif query == 'get_all_transient_ids_in_household':
        return HouseholdDevices(samples)
    elif query == "undecided_user_check":
        return AudienceCheck(samples)
    elif query == "undecided_user_audience":
        return AudienceGeneration(samples)
    elif query == "early_website_adopters":
        return EarlyAdopters(samples)