def _assert_expected_columns_in_dataframe(
join_keys: Set[str], entity_df_event_timestamp_col: str, entity_df: pandas.DataFrame
):
entity_df_columns = set(entity_df.columns.values)
expected_columns = join_keys.copy()
expected_columns.add(entity_df_event_timestamp_col)
missing_keys = expected_columns - entity_df_columns
if len(missing_keys) != 0:
raise errors.FeastEntityDFMissingColumnsError(expected_columns, missing_keys)
def _assert_expected_columns_in_bigquery(
join_keys: Set[str], entity_df_event_timestamp_col: str, entity_df_result
):
entity_columns = set()
for schema_field in entity_df_result.schema:
entity_columns.add(schema_field.name)
expected_columns = join_keys.copy()
expected_columns.add(entity_df_event_timestamp_col)
missing_keys = expected_columns - entity_columns
if len(missing_keys) != 0:
raise errors.FeastEntityDFMissingColumnsError(expected_columns, missing_keys)
