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: Handle complex Spark data types in SparkSource #3154

Merged
merged 2 commits into from
Aug 30, 2022

Conversation

rebstastic
Copy link
Contributor

@rebstastic rebstastic commented Aug 30, 2022

What this PR does / why we need it:
get_table_column_names_and_types of SparkSource returns non-string data type for complex Spark types.

For example, if a feature view contains a feature of type ArrayType(DoubleType(), False), current implementation would return a dictionary such as

{'type': 'array', 'elementType': 'double', 'containsNull': False}

With this fix, the returning data type would be array<double>.

This PR ensures that all Spark types are serialized as strings.

Which issue(s) this PR fixes:

Fixes #

Signed-off-by: Petra Rebernjak <[email protected]>
@rebstastic rebstastic changed the title Fix: Handle complex Spark data types in SparkSource fix: Handle complex Spark data types in SparkSource Aug 30, 2022
@achals
Copy link
Member

achals commented Aug 30, 2022

/ok-to-test

Signed-off-by: Petra Rebernjak <[email protected]>
@codecov-commenter
Copy link

codecov-commenter commented Aug 30, 2022

Codecov Report

Base: 67.11% // Head: 76.13% // Increases project coverage by +9.01% 🎉

Coverage data is based on head (b1164b2) compared to base (a14234d).
Patch coverage: 0.00% of modified lines in pull request are covered.

Additional details and impacted files
@@            Coverage Diff             @@
##           master    #3154      +/-   ##
==========================================
+ Coverage   67.11%   76.13%   +9.01%     
==========================================
  Files         175      211      +36     
  Lines       15907    17858    +1951     
==========================================
+ Hits        10676    13596    +2920     
+ Misses       5231     4262     -969     
Flag Coverage Δ
integrationtests 66.94% <0.00%> (-0.17%) ⬇️
unittests 58.28% <0.00%> (?)

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files Coverage Δ
...stores/contrib/spark_offline_store/spark_source.py 60.43% <0.00%> (+9.35%) ⬆️
...on/feast/infra/materialization/snowflake_engine.py 92.13% <0.00%> (-0.46%) ⬇️
.../online_stores/contrib/hbase_online_store/hbase.py 37.25% <0.00%> (ø)
...ib/trino_offline_store/test_config/manual_tests.py 33.33% <0.00%> (ø)
...offline_stores/contrib/spark_repo_configuration.py 100.00% <0.00%> (ø)
...s/contrib/trino_offline_store/tests/data_source.py 17.64% <0.00%> (ø)
...ores/contrib/mssql_offline_store/tests/__init__.py 100.00% <0.00%> (ø)
...s/contrib/mssql_offline_store/tests/data_source.py 46.93% <0.00%> (ø)
sdk/python/feast/infra/utils/hbase_utils.py 41.17% <0.00%> (ø)
...ffline_stores/contrib/trino_offline_store/trino.py 8.78% <0.00%> (ø)
... and 100 more

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.

☔ View full report at Codecov.
📢 Do you have feedback about the report comment? Let us know in this issue.

@rebstastic
Copy link
Contributor Author

/assign woop

@achals achals assigned achals and unassigned woop Aug 30, 2022
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

@feast-ci-bot
Copy link
Collaborator

[APPROVALNOTIFIER] This PR is APPROVED

This pull-request has been approved by: achals, rebstastic

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 5ddb83b into feast-dev:master Aug 30, 2022
adchia pushed a commit that referenced this pull request Aug 31, 2022
* Make sure data types are strings

Signed-off-by: Petra Rebernjak <[email protected]>

* Lint

Signed-off-by: Petra Rebernjak <[email protected]>

Signed-off-by: Petra Rebernjak <[email protected]>
adchia pushed a commit that referenced this pull request Sep 8, 2022
## [0.24.1](v0.24.0...v0.24.1) (2022-09-08)

### Bug Fixes

* Fix push API to respect feature view's already inferred entity types ([#3172](#3172)) ([5dba4a3](5dba4a3))
* Fix release workflow ([#3144](#3144)) ([c77a655](c77a655))
* Fix Shopify timestamp bug and add warnings to help with debugging entity registration ([#3191](#3191)) ([95066ea](95066ea))
* Handle complex Spark data types in SparkSource ([#3154](#3154)) ([0e13573](0e13573))
felixwang9817 pushed a commit that referenced this pull request Sep 20, 2022
# [0.25.0](v0.24.0...v0.25.0) (2022-09-20)

### Bug Fixes

* Broken Feature Service Link ([#3227](#3227)) ([e117082](e117082))
* Feature-server image is missing mysql dependency for mysql registry ([#3223](#3223)) ([ae37b20](ae37b20))
* Fix handling of TTL in Go server ([#3232](#3232)) ([f020630](f020630))
* Fix materialization when running on Spark cluster. ([#3166](#3166)) ([175fd25](175fd25))
* Fix push API to respect feature view's already inferred entity types ([#3172](#3172)) ([7c50ab5](7c50ab5))
* Fix release workflow ([#3144](#3144)) ([20a9dd9](20a9dd9))
* Fix Shopify timestamp bug and add warnings to help with debugging entity registration ([#3191](#3191)) ([de75971](de75971))
* Handle complex Spark data types in SparkSource ([#3154](#3154)) ([5ddb83b](5ddb83b))
* Local staging location provision ([#3195](#3195)) ([cdf0faf](cdf0faf))
* Remove bad snowflake offline store method ([#3204](#3204)) ([dfdd0ca](dfdd0ca))
* Remove opening file object when validating S3 parquet source ([#3217](#3217)) ([a906018](a906018))
* Snowflake config file search error ([#3193](#3193)) ([189afb9](189afb9))
* Update Snowflake Online docs ([#3206](#3206)) ([7bc1dff](7bc1dff))

### Features

* Add `to_remote_storage` functionality to `SparkOfflineStore` ([#3175](#3175)) ([2107ce2](2107ce2))
* Add ability to give boto extra args for registry config ([#3219](#3219)) ([fbc6a2c](fbc6a2c))
* Add health endpoint to py server ([#3202](#3202)) ([43222f2](43222f2))
* Add snowflake support for date & number with scale ([#3148](#3148)) ([50e8755](50e8755))
* Add tag kwarg to set Snowflake online store table path ([#3176](#3176)) ([39aeea3](39aeea3))
* Add workgroup to athena offline store config ([#3139](#3139)) ([a752211](a752211))
* Implement spark materialization engine ([#3184](#3184)) ([a59c33a](a59c33a))
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.

5 participants