Add endpoints property in TransformersModel

alvarobartt · Mar 8, 2024 · 08fabf6 · 08fabf6
1 parent 840a119
commit 08fabf6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/vertex_ai_huggingface_inference_toolkit/transformers.py b/src/vertex_ai_huggingface_inference_toolkit/transformers.py
@@ -191,6 +191,12 @@ def __init__(
  Union[aiplatform.Endpoint, aiplatform.PrivateEndpoint]
  ] = []
 
+ @property
+ def endpoints(
+ self,
+ ) -> Optional[List[Union[aiplatform.Endpoint, aiplatform.PrivateEndpoint]]]:
+ return self._endpoints
+
  def deploy(
  self,
  machine_type: Optional[str] = None,