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

fix: Update import logic to remove pyspark dependency from Snowflake Offline Store #3397

Merged
merged 4 commits into from
Dec 18, 2022

Conversation

beauchbum
Copy link
Contributor

@beauchbum beauchbum commented Dec 15, 2022

What this PR does / why we need it:

  • Moves the pyspark import inside of the relevant method so that the Snowflake offline store does not have a necessary dependency on pyspark
  • Uses typing.TYPE_CHECKING to do the imports while type checking, but not in runtime

Resulting behavior

  • The Snowflake Offline Store can be used without pyspark installed
    • However, when running an environment without pyspark, then calling to_spark_df raises a FeastExtrasDependencyImportError

Which issue(s) this PR fixes:
Fixes this issue

Signed-off-by: Ryan Beauchamp <ryan.beauchamp@ezcater.com>
Signed-off-by: Ryan Beauchamp <ryan.beauchamp@ezcater.com>
Signed-off-by: Ryan Beauchamp <ryan.beauchamp@ezcater.com>
@beauchbum
Copy link
Contributor Author

Went back and forth on this a lot to get things to work. Dealing with an optional import (pyspark) that needs to pass mypy type checking was difficult. I found a solution from this post.

Copy link
Member

@achals achals left a comment

Choose a reason for hiding this comment

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

/lgtm

Signed-off-by: Ryan Beauchamp <ryan.beauchamp@ezcater.com>
@feast-ci-bot
Copy link
Collaborator

[APPROVALNOTIFIER] This PR is APPROVED

This pull-request has been approved by: achals, beauchbum, sfc-gh-madkins

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:
  • OWNERS [achals,sfc-gh-madkins]

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

@sfc-gh-madkins
Copy link
Collaborator

/lgtm

@sfc-gh-madkins sfc-gh-madkins enabled auto-merge (squash) December 18, 2022 04:10
@feast-ci-bot feast-ci-bot merged commit cf073e6 into feast-dev:master Dec 18, 2022
felixwang9817 pushed a commit that referenced this pull request Jan 3, 2023
# [0.28.0](v0.27.0...v0.28.0) (2023-01-03)

### Bug Fixes

* Apply billing project when infer schema ([#3417](#3417)) ([4f9ad7e](4f9ad7e))
* Assertion condition when value is 0 ([#3401](#3401)) ([98a24a3](98a24a3))
* Enable registry caching in SQL Registry ([#3395](#3395)) ([2e57376](2e57376))
* Fix bug where SQL registry was incorrectly writing infra config around online stores ([#3394](#3394)) ([6bcf77c](6bcf77c))
* Get all columns with describe table method from RedshiftData-api ([#3377](#3377)) ([fd97254](fd97254))
* ODFV able to handle boolean pandas type ([#3384](#3384)) ([8f242e6](8f242e6))
* Remove PySpark dependency from Snowflake Offline Store ([#3388](#3388)) ([7b160c7](7b160c7))
* Specifies timeout in exception polling ([#3398](#3398)) ([c0ca7e4](c0ca7e4))
* Update import logic to remove `pyspark` dependency from Snowflake Offline Store ([#3397](#3397)) ([cf073e6](cf073e6))

### Features

* Add template for Github Codespaces ([#3421](#3421)) ([41c0537](41c0537))
* Adds description attribute for features/fields ([#3425](#3425)) ([26f4881](26f4881))
* Snowflake skip materialization if no table change ([#3404](#3404)) ([0ab3942](0ab3942))
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.

pyspark dependency breaks Snowflake offline store
4 participants