from metaflow import (  # type: ignore
    Flow,
    FlowSpec,
    Parameter,
    Run,
    metadata,
    namespace,
)

def get_runs_by_inference_date(inference_date: date) -> tuple[Run, Run]:
    """For a given inference date, return the relevant training and inference runs.

    :param model_enum: the model
    :param inference_date: the inference date to get prod runs for
    :return: (Training run, Inference run)
    """
    namespace(None)

    # Inference run
    inference_runs = Flow("Inference").runs(
        "project_branch:prod.stage",
        f"flow-code:{FLOW_CODE}",
    )
    inference_runs_on_date = [x for x in inference_runs if x.created_at.date() == inference_date]
    assert len(inference_runs_on_date) > 0, "no inference runs located"
    assert len(inference_runs_on_date) == 1, "multiple inference runs located"
    inference_run = inference_runs_on_date[0]

    training_runs = list(
        Flow("Training").runs(
            f"model-id:{inference_run.data.model_id}",  # type: ignore
        )
    )
    assert len(training_runs) > 0, "no training runs located"
    assert len(training_runs) == 1, "multiple training runs located"
    training_run = training_runs[0]

    return training_run, inference_run

error: Incompatible return value type (got "tuple[metaflow.client.core.Run, metaflow.client.core.Run]", expected "tuple[metaflow.Run, metaflow.Run]")  [return-value]

@kubernetes(image=KUBERNETES_IMAGE, node_selector=KUBERNETES_NODE_SELECTOR)

error: Unexpected keyword argument "node_selector" for "kubernetes"  [call-arg]