added init prepared statement call and updated parameters (#720)

logicalclocks · Jul 15, 2022 · 6caba0a · 6caba0a
1 parent 7582738
commit 6caba0a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python/hsfs/training_dataset.py b/python/hsfs/training_dataset.py
@@ -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
@@ -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):