One example, that's right. Another one:

    train_query = f"""
    SELECT 
      {{feature_names}},
      {",".join(labels)},
    ...
    """

and right at the top:

    cat_names = [
    ...
    ]