Function _upload_entity_df

sdk/python/feast/infra/offline_stores/bigquery.py:1400–1425

Uploads a Pandas entity dataframe into a BigQuery table and returns the resulting table

(
    client: Client,
    table_name: str,
    entity_df: Union[pd.DataFrame, str],
)

def _upload_entity_df(
    client: Client,
    table_name: str,
    entity_df: Union[pd.DataFrame, str],
) -> Table:
    """Uploads a Pandas entity dataframe into a BigQuery table and returns the resulting table"""
    job: Union[bigquery.job.query.QueryJob, bigquery.job.load.LoadJob]

    if isinstance(entity_df, str):
        job = client.query(f"CREATE TABLE `{table_name}` AS ({entity_df})")

    elif isinstance(entity_df, pd.DataFrame):
        # Drop the index so that we don't have unnecessary columns
        entity_df.reset_index(drop=True, inplace=True)
        job = client.load_table_from_dataframe(entity_df, table_name)
    else:
        raise InvalidEntityType(type(entity_df))

    block_until_done(client, job)

    # Ensure that the table expires after some time
    table = client.get_table(table=table_name)
    table.expires = _utc_now() + timedelta(minutes=30)
    client.update_table(table, ["expires"])

    return table


def _get_entity_schema(