def METHOD_NAME(cls):
    METHOD_NAME = super().METHOD_NAME()
    METHOD_NAME.update({
        'saved_model_dir': PathField(is_directory=True, optional=True, description='Path to saved model directory'),
        'device': StringField(
            choices=('cpu', 'gpu'), default='cpu', optional=True, description="Device name: cpu or gpu"),
        'inputs': BaseField(optional=True, description="Inputs."),
        'output_names': ListField(
            allow_empty=False, optional=True, value_type=StringField(), description="Output names."
        )
    })
    return METHOD_NAME