openshift-python-wrapper 10.0.62__tar.gz → 10.0.63__tar.gz

{openshift_python_wrapper-10.0.62 → openshift_python_wrapper-10.0.63}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: openshift-python-wrapper
-Version: 10.0.62
+Version: 10.0.63
 Summary: Wrapper around https://github.com/kubernetes-client/python
 Home-page: https://github.com/RedHatQE/openshift-python-wrapper
 License: Apache-2.0

openshift_python_wrapper-10.0.63/ocp_resources/inference_service.py

@@ -0,0 +1,50 @@
+from typing import Optional, Dict, Any
+
+from ocp_resources.resource import NamespacedResource, MissingRequiredArgumentError
+
+
+class InferenceService(NamespacedResource):
+    """
+    https://github.com/kserve/kserve/blob/master/pkg/apis/serving/v1beta1/inference_service.go
+    """
+
+    api_group: str = NamespacedResource.ApiGroup.SERVING_KSERVE_IO
+
+    def __init__(
+        self,
+        predictor: Optional[Dict[str, Any]] = None,
+        explainer: Optional[Dict[str, Any]] = None,
+        transformer: Optional[Dict[str, Any]] = None,
+        **kwargs,
+    ):
+        """
+        InferenceService object
+
+        Args:
+            predictor (Optional[Dict[str, Any]], mandatory if yaml_file is not provided): Defines the model serving spec.
+            explainer (Optional[Dict[str, Any]]): Defines the model explanation service spec.
+            transformer (Optional[Dict[str, Any]]): Defines the pre/post processing before and after the predictor call.
+        """
+        super().__init__(
+            **kwargs,
+        )
+        self.predictor = predictor
+        self.explainer = explainer
+        self.transformer = transformer
+
+    def to_dict(self) -> None:
+        super().to_dict()
+        if not self.yaml_file:
+            if not self.predictor:
+                raise MissingRequiredArgumentError(argument="predictor")
+
+            self.res["spec"] = {}
+            _spec = self.res["spec"]
+
+            _spec["predictor"] = self.predictor
+
+            if self.explainer:
+                _spec["explainer"] = self.explainer
+
+            if self.transformer:
+                _spec["transformer"] = self.transformer

{openshift_python_wrapper-10.0.62 → openshift_python_wrapper-10.0.63}/pyproject.toml

@@ -19,7 +19,7 @@ warn_unused_ignores = true
 
 [tool.poetry]
 name = "openshift-python-wrapper"
-version = "10.0.62"
+version = "10.0.63"
 description = "Wrapper around https://github.com/kubernetes-client/python"
 authors = ["Meni Yakove <myakove@gmail.com>", "Ruth Netser <rnetser@gmail.com>"]
 readme = "README.md"