Skip to content
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

Merged
merged 11 commits into from
Mar 12, 2021

Conversation

jklegar
Copy link
Collaborator

@jklegar jklegar commented Mar 5, 2021

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?:

NONE

@jklegar
Copy link
Collaborator Author

jklegar commented Mar 5, 2021

btw this is stacked on #1364 but I don't think GH PR's support stacked diffs

query: Optional[str],
):
if (table_ref is None) == (query is None):
raise Exception("Exactly one of table_ref and query should be specified")
Copy link
Member

Choose a reason for hiding this comment

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

sdk/python/setup.py Outdated Show resolved Hide resolved

# 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 = (
Copy link
Member

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

@woop woop self-requested a review March 9, 2021 01:05
sdk/python/feast/feature_store.py Outdated Show resolved Hide resolved

class BigQueryOfflineStore(OfflineStore):
"""
BigQueryOfflineStore is a non-user-facing object used for all interaction between Feast and BigQuery.
Copy link
Member

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.

@jklegar jklegar added approved and removed approved labels Mar 11, 2021
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]>
@woop
Copy link
Member

woop commented Mar 12, 2021

/lgtm

@feast-ci-bot
Copy link
Collaborator

[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 /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

@feast-ci-bot feast-ci-bot merged commit 8ada7ae into feast-dev:master Mar 12, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants