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)