Visitar URL original
feat: Offline store update pull_all_from_table_or_query to make timestampfield optional by haoxu0 · Pull Request #5281 · feast-dev/feast · GitHub
Skip to content

feat: Offline store update pull_all_from_table_or_query to make timestampfield optional #5281

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
franciscojavierarceo merged 20 commits into from
Apr 20, 2025
Merged

feat: Offline store update pull_all_from_table_or_query to make timestampfield optional #5281

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
112014e
Update offline store pull all API date field optional
haoxu0 Apr 18, 2025
c75c264
Update offline store pull all API date field optional
haoxu0 Apr 18, 2025
b62fd1a
update
haoxu0 Apr 18, 2025
8b5bdae
update
haoxu0 Apr 18, 2025
e92d015
update
haoxu0 Apr 18, 2025
4765d8f
fix test
haoxu0 Apr 18, 2025
0bbdbbc
update source read node
haoxu0 Apr 18, 2025
0336318
fix linting
haoxu0 Apr 18, 2025
ac03a75
fix linting
haoxu0 Apr 18, 2025
0033b9c
fix linting
haoxu0 Apr 18, 2025
743a490
fix linting
haoxu0 Apr 18, 2025
4285b5f
fix linting
haoxu0 Apr 18, 2025
618c4b8
fix test
haoxu0 Apr 18, 2025
afcaf9c
fix test
haoxu0 Apr 18, 2025
8e02747
fix test
haoxu0 Apr 18, 2025
71daec2
fix test
haoxu0 Apr 18, 2025
23a0fb4
fix test
haoxu0 Apr 18, 2025
bcf1d7a
fix test
haoxu0 Apr 18, 2025
89ad0da
fix test
haoxu0 Apr 18, 2025
cdae95b
fix test
haoxu0 Apr 18, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix test 
Signed-off-by: HaoXuAI <sduxuhao@gmail.com>
  • Loading branch information
@haoxu0
haoxu0 committed Apr 18, 2025
commit 23a0fb41be08afa06313aa2468885f4917500922
6 changes: 3 additions & 3 deletions sdk/python/feast/infra/offline_stores/offline_utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -274,7 +274,7 @@ def get_timestamp_filter_sql(
timestamp_field: Optional[str] = DEFAULT_ENTITY_DF_EVENT_TIMESTAMP_COL,
date_partition_column: Optional[str] = None,
tz: Optional[timezone] = None,
cast_style: Literal["timestamp_func", "timestamptz", "raw"] = "timestamp_func",
cast_style: Literal["timestamp_func", "timestamp", "timestamptz", "raw"] = "timestamp_func",
date_time_separator: str = "T",
quote_fields: bool = True,
) -> str:
Expand Down Expand Up @@ -314,8 +314,8 @@ def format_casted_ts(val: Union[str, datetime]) -> str:

if cast_style == "timestamp_func":
return f"TIMESTAMP('{val_str}')"
elif cast_style == "timestamptz":
return f"'{val_str}'::timestamptz"
elif cast_style in ("timestamptz", "timestamp"):
return f"'{val_str}'::{cast_style}"
else:
return f"'{val_str}'"

Expand Down
2 changes: 1 addition & 1 deletion sdk/python/feast/infra/offline_stores/redshift.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -175,7 +175,7 @@ def pull_all_from_table_or_query(
s3_resource = aws_utils.get_s3_resource(config.offline_store.region)

timestamp_filter = get_timestamp_filter_sql(
start_date, end_date, timestamp_field, tz=timezone.utc
start_date, end_date, timestamp_field, tz=timezone.utc, cast_style="timestamp"
)

query = f"""
Expand Down