Skip to content

Conversation

@khorshuheng
Copy link
Collaborator

What this PR does / why we need it:
The bigquery table corresponds to the entity source should be partitioned in order to achieve better performance for historical retrieval job.

Which issue(s) this PR fixes:

Fixes #

Does this PR introduce a user-facing change?:


@feast-ci-bot
Copy link
Collaborator

[APPROVALNOTIFIER] This PR is APPROVED

This pull-request has been approved by: khorshuheng

The full list of commands accepted by this bot can be found here.

The pull request process is described here

Details 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

@khorshuheng khorshuheng force-pushed the partitioned-entity-source branch 2 times, most recently from 8d82a81 to 98411f4 Compare June 8, 2022 03:55
Table will expire in 1 day.
Returns BigQuerySource with reference to created table.
"""
from google.cloud import bigquery
Copy link
Contributor

Choose a reason for hiding this comment

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

what is the rationale for importing inside a func ?

I guess this is not expected to run in user's env and thus this dependency should be present in env ?

Is this method expected to be called rarely ?

)


def stage_entities_to_bq_with_partition(
Copy link
Contributor

Choose a reason for hiding this comment

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

can we avoid duplication by adding partition as an optional argument in existing feast.staging.entities file ?

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

The issue is that, we would have to make this change in the feast repository, and we are having issue publishing Python SDK for branch 0.9 at the moment.

@khorshuheng khorshuheng force-pushed the partitioned-entity-source branch from 98411f4 to 488e734 Compare June 8, 2022 05:40
Signed-off-by: shuheng <khor.heng@gojek.com>

Co-authored-by: shuheng <khor.heng@gojek.com>
@khorshuheng khorshuheng force-pushed the partitioned-entity-source branch 2 times, most recently from 950e689 to ae5715a Compare June 8, 2022 07:43
@khorshuheng khorshuheng merged commit 67567fd into feast-dev:master Jun 8, 2022
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