Skip to content

Commit 023318e

Browse files
authored
[FSTORE-1857] Enable get feature vector for spine group v4.5 (#660) (#662)
* [FSTORE-1857] Enable get feature vector for spine group (#660) * Enable get feature vector for spine group * style fix * style fix * pump version to 4.5.0-rc5
1 parent 71f41cf commit 023318e

10 files changed

Lines changed: 26 additions & 11 deletions

File tree

java/beam/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
<parent>
66
<artifactId>hsfs-parent</artifactId>
77
<groupId>com.logicalclocks</groupId>
8-
<version>4.5.0-RC4</version>
8+
<version>4.5.0-RC5</version>
99
</parent>
1010
<modelVersion>4.0.0</modelVersion>
1111

java/flink/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
<parent>
66
<artifactId>hsfs-parent</artifactId>
77
<groupId>com.logicalclocks</groupId>
8-
<version>4.5.0-RC4</version>
8+
<version>4.5.0-RC5</version>
99
</parent>
1010
<modelVersion>4.0.0</modelVersion>
1111

java/hsfs/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
<parent>
66
<artifactId>hsfs-parent</artifactId>
77
<groupId>com.logicalclocks</groupId>
8-
<version>4.5.0-RC4</version>
8+
<version>4.5.0-RC5</version>
99
</parent>
1010
<modelVersion>4.0.0</modelVersion>
1111

java/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
<groupId>com.logicalclocks</groupId>
88
<artifactId>hsfs-parent</artifactId>
99
<packaging>pom</packaging>
10-
<version>4.5.0-RC4</version>
10+
<version>4.5.0-RC5</version>
1111
<modules>
1212
<module>hsfs</module>
1313
<module>spark</module>

java/spark/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
<parent>
2323
<artifactId>hsfs-parent</artifactId>
2424
<groupId>com.logicalclocks</groupId>
25-
<version>4.5.0-RC4</version>
25+
<version>4.5.0-RC5</version>
2626
</parent>
2727
<modelVersion>4.0.0</modelVersion>
2828

python/hopsworks_common/version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,4 +14,4 @@
1414
# limitations under the License.
1515
#
1616

17-
__version__ = "4.5.0rc4"
17+
__version__ = "4.5.0rc5"

python/hsfs/core/vector_server.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1931,6 +1931,7 @@ def _all_feature_groups_online(self) -> bool:
19311931
"""True if all feature groups in the feature view are online."""
19321932
if self.__all_feature_groups_online is None:
19331933
self.__all_feature_groups_online = all(
1934-
fg.online_enabled for fg in self._parent_feature_groups
1934+
fg.online_enabled
1935+
for fg in self._parent_feature_groups if fg.id not in self._skip_fg_ids
19351936
)
19361937
return self.__all_feature_groups_online

python/hsfs/feature_group.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1907,6 +1907,11 @@ def feature_store(self) -> feature_store_mod.FeatureStore:
19071907
def feature_store(self, feature_store: feature_store_mod.FeatureStore) -> None:
19081908
self._feature_store = feature_store
19091909

1910+
@property
1911+
def id(self) -> Optional[int]:
1912+
"""Feature group id."""
1913+
return self._id
1914+
19101915
@property
19111916
def name(self) -> Optional[str]:
19121917
"""Name of the feature group."""
@@ -4584,7 +4589,8 @@ def __init__(
45844589
great_expectations.core.ExpectationSuite,
45854590
]
45864591
] = None,
4587-
online_enabled: bool = False,
4592+
# spine groups are online enabled by default such that feature_view.get_feature_vector can be used
4593+
online_enabled: bool = True,
45884594
href: Optional[str] = None,
45894595
online_topic_name: Optional[str] = None,
45904596
topic_name: Optional[str] = None,
@@ -4773,4 +4779,5 @@ def to_dict(self) -> Dict[str, Any]:
47734779
"spine": True,
47744780
"topicName": self.topic_name,
47754781
"deprecated": self.deprecated,
4782+
"onlineEnabled": self._online_enabled,
47764783
}

python/hsfs/feature_view.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4248,14 +4248,21 @@ def feature_logging(self) -> Optional[FeatureLogging]:
42484248
self._feature_logging = self._feature_view_engine.get_feature_logging(self)
42494249
return self._feature_logging
42504250

4251+
def _get_spine_fg_ids(self) -> List[feature_group.SpineGroup]:
4252+
return [fg.id for fg in self.query.featuregroups if isinstance(fg, feature_group.SpineGroup)]
4253+
4254+
def _get_skip_fg_ids(self) -> Set[int]:
4255+
embedding_fg_ids = [fg.id for fg in self._get_embedding_fgs()]
4256+
return set(embedding_fg_ids + self._get_spine_fg_ids())
4257+
42514258
@property
42524259
def _vector_server(self) -> vector_server.VectorServer:
42534260
if not self.__vector_server:
42544261
self.__vector_server = vector_server.VectorServer(
42554262
feature_store_id=self._featurestore_id,
42564263
features=self._features,
42574264
serving_keys=self._serving_keys,
4258-
skip_fg_ids=set([fg.id for fg in self._get_embedding_fgs()]),
4265+
skip_fg_ids=self._get_skip_fg_ids(),
42594266
feature_view_name=self._name,
42604267
feature_view_version=self._version,
42614268
feature_store_name=self._feature_store_name,
@@ -4269,7 +4276,7 @@ def _batch_scoring_server(self) -> vector_server.VectorServer:
42694276
feature_store_id=self._featurestore_id,
42704277
features=self._features,
42714278
serving_keys=self._serving_keys,
4272-
skip_fg_ids=set([fg.id for fg in self._get_embedding_fgs()]),
4279+
skip_fg_ids=self._get_skip_fg_ids(),
42734280
feature_view_name=self._name,
42744281
feature_view_version=self._version,
42754282
feature_store_name=self._feature_store_name,

utils/java/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
<groupId>com.logicalclocks</groupId>
77
<artifactId>hsfs-utils</artifactId>
8-
<version>4.5.0-RC4</version>
8+
<version>4.5.0-RC5</version>
99

1010
<properties>
1111
<hops.version>3.2.0.0-SNAPSHOT</hops.version>

0 commit comments

Comments
 (0)