orient_express 0.3.2__tar.gz → 0.3.3__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: orient_express
-Version: 0.3.2
+Version: 0.3.3
 Summary: A library to simplify model deployment to Vertex AI
 Author: Alexey Zankevich
 Author-email: alex.zankevich@shiftsmart.com

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

@@ -44,15 +44,22 @@ class ModelExpress:
         else:
             self.endpoint_name = endpoint_name
 
+        self._vertex_initialized = False
+
     def colab_auth(self):
         from google.colab import auth
 
         auth.authenticate_user()
 
     def _vertex_init(self):
-        aiplatform.init(project=self.project_name, location=self.region)
+        if not self._vertex_initialized:
+            aiplatform.init(project=self.project_name, location=self.region)
+            self._vertex_initialized = True
 
     def get_latest_vertex_model(self, model_name):
+        """If there are a few models with the same name, load the most recent one.
+        It's highly recommended to keep only 1 model with the same name to avoid the confusion
+        """
         self._vertex_init()
 
         # Search for models with the specified display name
@@ -153,6 +160,8 @@ class ModelExpress:
         )
 
     def remote_predict(self, input_df):
+        self._vertex_init()
+
         if not self.endpoint:
             endpoint = self.get_endpoint()
             if not endpoint:
@@ -166,6 +175,8 @@ class ModelExpress:
         return predictions.predictions
 
     def local_predict(self, input_df):
+        self._vertex_init()
+
         if not self.model:
             self.load_model_from_registry()
 
@@ -178,14 +189,16 @@ class ModelExpress:
         return self.model.predict_proba(input_df)
 
     def load_model_from_registry(self):
+        self._vertex_init()
+
+        vertex_model = self.get_latest_vertex_model(self.model_name)
+
         if self.model_version:
+            # reload the model using a specific model version
             vertex_model = aiplatform.Model(
-                model_name=self.model_name, version=self.model_version
+                model_name=vertex_model.resource_name, version=str(self.model_version)
             )
 
-        else:
-            vertex_model = self.get_latest_vertex_model(self.model_name)
-
         if not vertex_model:
             raise Exception(f"Model '{self.model_name}' not found in the registry.")
 

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "orient_express"
-version = "0.3.2"
+version = "0.3.3"
 description = "A library to simplify model deployment to Vertex AI"
 authors = ["Alexey Zankevich <alex.zankevich@shiftsmart.com>"]
 readme = "README.md"