Merge pull request #615 from basetenlabs/bump-version-0.6.4

Release 0.6.4

Author: bolasim

.github/workflows/create_release_pr.yml

@@ -47,7 +47,6 @@ jobs:
         CURR_VERSION=$(curl https://pypi.org/pypi/truss/json | jq ".info.version")
         PR_BODY="Updating Truss from [$CURR_VERSION](https://pypi.org/pypi/truss/json) to $INPUT_VERSION."
         PR_URL=$(gh pr create --base release --head refs/heads/bump-version-$INPUT_VERSION --title "Release $INPUT_VERSION" --body "$PR_BODY")
-        gh pr merge $PR_URL --auto --merge
       env:
         INPUT_VERSION: ${{ github.event.inputs.version }}
         GH_TOKEN: ${{ secrets.BASETENBOT_GITHUB_TOKEN }}

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truss"
-version = "0.6.3"
+version = "0.6.4"
 description = "A seamless bridge from model development to model delivery"
 license = "MIT"
 readme = "README.md"

truss/contexts/image_builder/cache_warmer.py

@@ -23,8 +23,11 @@ def download_file(
             storage_client = storage.Client.from_service_account_json(key_file)
             bucket = storage_client.bucket(repo_name)
             blob = bucket.blob(file_name)
+            dst_file = Path(f"{cache_dir}/{file_name}")
+            if not dst_file.parent.exists():
+                dst_file.parent.mkdir(parents=True)
             # Download the blob to a file
-            blob.download_to_filename(f"{cache_dir}/{file_name}")
+            blob.download_to_filename(dst_file)
         except Exception as e:
             raise RuntimeError(f"Failure downloading file from GCS: {e}")
     else: