orient_express 2.1.2__tar.gz → 2.1.3__tar.gz

{orient_express-2.1.2 → orient_express-2.1.3}/PKG-INFO

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

{orient_express-2.1.2 → orient_express-2.1.3}/orient_express/vertex.py

@@ -76,13 +76,15 @@ class VertexModel:
         predictions = self.endpoint.predict(instances=instances, parameters=parameters)
         return predictions.predictions
 
-    def get_local_predictor(self, device: str = "cpu"):
+    def get_local_predictor(self, device: str = "cpu", force_download: bool = False):
         dir = os.path.join(ARTIFACT_DIR, self.model_name + "-" + str(self.version))
-        self.download_artifacts(dir)
+        self.download_artifacts(dir, force_download=force_download)
         return get_predictor(dir, device)
 
-    def download_artifacts(self, dir: str):
-        download_artifacts(dir, self.vertex_model.gca_resource.artifact_uri)
+    def download_artifacts(self, dir: str, force_download: bool = True):
+        download_artifacts(
+            dir, self.vertex_model.gca_resource.artifact_uri, force_download
+        )
 
 
 def vertex_init(project_name: str, region: str):
@@ -92,18 +94,17 @@ def vertex_init(project_name: str, region: str):
         _vertex_initialized = True
 
 
-def download_artifacts(dir: str, artifact_uri: str):
+def download_artifacts(dir: str, artifact_uri: str, force_download: bool = True):
     storage_client = storage.Client()
-
     bucket_name, artifact_path = artifact_uri.replace("gs://", "").split("/", 1)
     bucket = storage_client.bucket(bucket_name)
-
     os.makedirs(dir, exist_ok=True)
-
     blobs = bucket.list_blobs(prefix=artifact_path)
     for blob in blobs:
         filename = blob.name.split("/")[-1]
         download_path = os.path.join(dir, filename)
+        if not force_download and os.path.exists(download_path):
+            continue
         blob.download_to_filename(download_path)
 
 

{orient_express-2.1.2 → orient_express-2.1.3}/pyproject.toml

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