Skip to content
Merged
Prev Previous commit
Next Next commit
fix test
Signed-off-by: HaoXuAI <sduxuhao@gmail.com>
  • Loading branch information
HaoXuAI committed Apr 18, 2025
commit 89ad0da9fbd1295d2a4f38c2568d52faf6f599e6
6 changes: 5 additions & 1 deletion sdk/python/feast/infra/offline_stores/bigquery.py
Original file line number Diff line number Diff line change
Expand Up @@ -208,7 +208,11 @@ def pull_all_from_table_or_query(
+ [timestamp_field]
)
timestamp_filter = get_timestamp_filter_sql(
Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

major change: get the timestamp filter from get_timestamp_filter_sql

start_date, end_date, timestamp_field, quote_fields=False, cast_style="timestamp_func"
start_date,
end_date,
timestamp_field,
quote_fields=False,
cast_style="timestamp_func",
)
query = f"""
SELECT {field_string}
Expand Down
3 changes: 2 additions & 1 deletion sdk/python/feast/infra/offline_stores/offline_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -290,7 +290,8 @@ def get_timestamp_filter_sql(
date_partition_column: optional partition column (for pruning)
tz: optional timezone for datetime inputs
cast_style: one of:
- "timestamp_func": TIMESTAMP('...') → Snowflake, BigQuery, Athena
- "timestamp": TIMESTAMP '...' → Common Sql engine Snowflake, Redshift etc.
- "timestamp_func": TIMESTAMP('...') → BigQuery, Couchbase etc.
- "timestamptz": '...'::timestamptz → PostgreSQL
- "raw": '...' → no cast, string only
date_time_separator: separator for datetime strings (default is "T")
Expand Down
Loading