def handle_register()

in lambda/api/lambda_register.py [0:0]


def handle_register(endpoint_name: str, strategy: str, epsilon: float, warmup: int):
    endpoint_variants = get_endpoint_variants(endpoint_name)
    response = exp_metrics.create_variant_metrics(
        endpoint_name=endpoint_name,
        endpoint_variants=endpoint_variants,
        strategy=strategy,
        epsilon=epsilon,
        warmup=warmup,
    )
    result = {
        "endpoint_name": endpoint_name,
        "endpoint_variants": endpoint_variants,
        "strategy": strategy,
        "epsilon": epsilon,
        "warmup": warmup,
    }
    if "Attributes" not in response:
        return result, 201
    return result, 200