orient_express 0.2.5__tar.gz → 0.3.2__tar.gz

{orient_express-0.2.5 → orient_express-0.3.2}/PKG-INFO

@@ -1,16 +1,15 @@
 Metadata-Version: 2.1
 Name: orient_express
-Version: 0.2.5
+Version: 0.3.2
 Summary: A library to simplify model deployment to Vertex AI
 Author: Alexey Zankevich
 Author-email: alex.zankevich@shiftsmart.com
-Requires-Python: >=3.9
+Requires-Python: >=3.9,<3.13
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: google-cloud-aiplatform
 Requires-Dist: google-cloud-storage
 Requires-Dist: pandas

{orient_express-0.2.5 → orient_express-0.3.2}/orient_express/model_wrapper.py

@@ -16,7 +16,9 @@ class ModelExpress:
         model=None,
         region="us-central1",
         serialized_model_path="model.joblib",
-        docker_image_uri="us-docker.pkg.dev/vertex-ai/prediction/xgboost-cpu.1-7:latest",
+        serving_container_image_uri="us-west1-docker.pkg.dev/shiftsmart-api/orient-express/xgboost-scikit-learn:latest",
+        serving_container_predict_route="/v1/models/orient-express-model:predict",
+        serving_container_health_route="/v1/models/orient-express-model",
         endpoint_name=None,
         machine_type="n1-standard-4",
         min_replica_count=1,
@@ -29,14 +31,16 @@ class ModelExpress:
         self.project_name = project_name
         self.bucket_name = bucket_name
         self.serialized_model_path = serialized_model_path
-        self.docker_image_uri = docker_image_uri
+        self.serving_container_image_uri = serving_container_image_uri
+        self.serving_container_predict_route = serving_container_predict_route
+        self.serving_container_health_route = serving_container_health_route
         self.machine_type = machine_type
         self.min_replica_count = min_replica_count
         self.max_replica_count = max_replica_count
         self.endpoint = None
 
         if not endpoint_name:
-            self.endpoint_name = f"model-xpress-{model_name}"
+            self.endpoint_name = f"orient-express-{model_name}"
         else:
             self.endpoint_name = endpoint_name
 
@@ -106,7 +110,9 @@ class ModelExpress:
             display_name=self.model_name,
             artifact_uri=artifact_uri,
             parent_model=parent_model,
-            serving_container_image_uri=self.docker_image_uri,
+            serving_container_image_uri=self.serving_container_image_uri,
+            serving_container_health_route=self.serving_container_health_route,
+            serving_container_predict_route=self.serving_container_predict_route,
             sync=True,
         )
 
@@ -165,6 +171,12 @@ class ModelExpress:
 
         return self.model.predict(input_df)
 
+    def local_predict_proba(self, input_df):
+        if not self.model:
+            self.load_model_from_registry()
+
+        return self.model.predict_proba(input_df)
+
     def load_model_from_registry(self):
         if self.model_version:
             vertex_model = aiplatform.Model(
@@ -193,5 +205,4 @@ class ModelExpress:
             blob.download_to_filename(artifact_path)
 
     def df_to_features(self, df: pd.DataFrame):
-        #
         return df.to_dict(orient="records")

{orient_express-0.2.5 → orient_express-0.3.2}/pyproject.toml

@@ -1,12 +1,12 @@
 [tool.poetry]
 name = "orient_express"
-version = "0.2.5"
+version = "0.3.2"
 description = "A library to simplify model deployment to Vertex AI"
 authors = ["Alexey Zankevich <alex.zankevich@shiftsmart.com>"]
 readme = "README.md"
 
 [tool.poetry.dependencies]
-python = ">=3.9"
+python = ">=3.9,<3.13"
 google-cloud-aiplatform = "*"
 google-cloud-storage = "*"
 pandas = "*"
@@ -17,6 +17,12 @@ pytest = "8.3.3"
 scikit-learn = "1.5.2"
 xgboost = "2.1.2"
 
+[tool.poetry.group.inference.dependencies]
+scikit-learn = "1.5.2"
+kserve = "0.14.0"
+python-json-logger = "2.0.7"
+gcsfs = "^2024.10.0"
+
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"