orient_express 0.2.4__tar.gz → 0.3.1__tar.gz

{orient_express-0.2.4 → orient_express-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: orient_express
-Version: 0.2.4
+Version: 0.3.1
 Summary: A library to simplify model deployment to Vertex AI
 Author: Alexey Zankevich
 Author-email: alex.zankevich@shiftsmart.com
@@ -14,8 +14,6 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: google-cloud-aiplatform
 Requires-Dist: google-cloud-storage
 Requires-Dist: pandas
-Requires-Dist: scikit-learn (>=1.5.2,<2.0.0)
-Requires-Dist: xgboost (>=2.1.2,<3.0.0)
 Description-Content-Type: text/markdown
 
 # Orient Express

{orient_express-0.2.4 → orient_express-0.3.1}/orient_express/model_wrapper.py

@@ -165,6 +165,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(

{orient_express-0.2.4 → orient_express-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "orient_express"
-version = "0.2.4"
+version = "0.3.1"
 description = "A library to simplify model deployment to Vertex AI"
 authors = ["Alexey Zankevich <alex.zankevich@shiftsmart.com>"]
 readme = "README.md"
@@ -10,12 +10,12 @@ python = ">=3.9"
 google-cloud-aiplatform = "*"
 google-cloud-storage = "*"
 pandas = "*"
-scikit-learn = "^1.5.2"
-xgboost = "^2.1.2"
 
 [tool.poetry.dev-dependencies]
 black = "24.10.0"
 pytest = "8.3.3"
+scikit-learn = "1.5.2"
+xgboost = "2.1.2"
 
 [build-system]
 requires = ["poetry-core"]