Skip to content

Commit

Permalink
added init prepared statement call and updated parameters (logicalclo…
Browse files Browse the repository at this point in the history
  • Loading branch information
tdoehmen authored and kennethmhc committed Nov 16, 2022
1 parent df5bc2a commit 9be3b29
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions python/hsfs/training_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -772,7 +772,9 @@ def get_serving_vector(
`list` List of feature values related to provided primary keys, ordered according to positions of this
features in training dataset query.
"""
return self._vector_server.get_feature_vector(entry, external)
if self._vector_server.prepared_statements is None:
self.init_prepared_statement(None, external)
return self._vector_server.get_feature_vector(entry)

def get_serving_vectors(
self, entry: Dict[str, List[Any]], external: Optional[bool] = None
Expand All @@ -792,7 +794,9 @@ def get_serving_vectors(
`List[list]` List of lists of feature values related to provided primary keys, ordered according to
positions of this features in training dataset query.
"""
return self._vector_server.get_feature_vectors(entry, external)
if self._vector_server.prepared_statements is None:
self.init_prepared_statement(None, external)
return self._vector_server.get_feature_vectors(entry)

@property
def label(self):
Expand Down

0 comments on commit 9be3b29

Please sign in to comment.