-
Notifications
You must be signed in to change notification settings - Fork 998
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.
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
Download BigQuery table to pyarrow table for python-based ingestion flow #1366
Conversation
btw this is stacked on #1364 but I don't think GH PR's support stacked diffs |
sdk/python/feast/big_query_source.py
Outdated
query: Optional[str], | ||
): | ||
if (table_ref is None) == (query is None): | ||
raise Exception("Exactly one of table_ref and query should be specified") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
https://docs.python.org/3/library/exceptions.html#ValueError is preferred I think.
sdk/python/feast/offline_store.py
Outdated
|
||
# if we have mapped fields, use the original field names in the call to BigQuery | ||
event_timestamp_column = feature_view.inputs.event_timestamp_column | ||
fields = ( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can we please add a few comments to this code so that its easier for a reader to parse
579074d
to
8bcd3ed
Compare
35bccce
to
08bb0a2
Compare
sdk/python/feast/offline_store.py
Outdated
|
||
class BigQueryOfflineStore(OfflineStore): | ||
""" | ||
BigQueryOfflineStore is a non-user-facing object used for all interaction between Feast and BigQuery. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think we can maybe just leave out this comment and inherit from OfflineStore for now.
Signed-off-by: Jacob Klegar <[email protected]>
Signed-off-by: Jacob Klegar <[email protected]>
Signed-off-by: Jacob Klegar <[email protected]>
Signed-off-by: Jacob Klegar <[email protected]>
Signed-off-by: Jacob Klegar <[email protected]>
Signed-off-by: Jacob Klegar <[email protected]>
Signed-off-by: Jacob Klegar <[email protected]>
Signed-off-by: Jacob Klegar <[email protected]>
Signed-off-by: Jacob Klegar <[email protected]>
Signed-off-by: Jacob Klegar <[email protected]>
Signed-off-by: Jacob Klegar <[email protected]>
47ee560
to
85465a1
Compare
/lgtm |
[APPROVALNOTIFIER] This PR is APPROVED This pull-request has been approved by: jklegar, woop The full list of commands accepted by this bot can be found here. The pull request process is described here
Needs approval from an approver in each of these files:
Approvers can indicate their approval by writing |
What this PR does / why we need it: Implements downloading a BQ table to an in-memory pyarrow table, which will be part of the python-based ingestion flow
Which issue(s) this PR fixes:
Fixes #
Does this PR introduce a user-facing change?: