test_overrides.py

OVERRIDE_INPUTS = {
    'Pendrokar/xVASynth-TTS': {
        1: 'x_ex04', #fine-tuned voice model name
        3: 1.0, #pacing/duration - Gradio Slider issue: takes min. rather than value
    },
}

def _get_param_examples(parameters):
    # named or unnamed parameters
    try:
        param_name = parameters[0]['parameter_name']
        # success => named params, use dict
        example_inputs = {}
    except:
        # unnamed params, use list
        example_inputs = []
        pass

    for param_info in parameters:


        param_name = ''
        param_default_value = param_info['example_input']
        try:
            # named params
            param_name = param_info['parameter_name']
            param_default_value = param_info['parameter_default']
        except:
            # unnamed params
            pass

        param_value = None
        if (
            param_info['component'] == 'Radio'
            or param_info['component'] == 'Dropdown'
            or param_info['component'] == 'Audio'
            or param_info['python_type']['type'] == 'str'
        ):
            param_value = str(param_default_value)
        elif param_info['python_type']['type'] == 'int':
            param_value = int(param_default_value)
        elif param_info['python_type']['type'] == 'float':
            param_value = float(param_default_value)
        elif param_info['python_type']['type'] == 'bool':
            param_value = bool(param_default_value)

        if (param_name != ''):
            # named param
            example_inputs[param_info['parameter_name']] = param_value
        else:
            # just append unnamed param and hope
            example_inputs.append(param_value)

    return example_inputs

def _override_params(inputs, modelname):
    try:
        for key,value in OVERRIDE_INPUTS[modelname].items():
            # if override keys are integers, make the dict into a list
            if (
                (type(inputs) is dict)
                and (type(key) is int)
            ):
                print("Converting unnamed override params to List")
                inputs = list(inputs.values())

            inputs[key] = value
        print(f"{modelname}: Default inputs overridden by Arena")
    except:
        pass

    return inputs