Fix Fallback Value

square · Jul 22, 2024 · 0558f0a · 0558f0a
1 parent f8c849a
commit 0558f0a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/block_cascade/executors/vertex/executor.py b/block_cascade/executors/vertex/executor.py
@@ -203,7 +203,16 @@ def create_job(self) -> VertexJob:
                 #
                 # TODO: This should be generalized to accept any storage block.
                 #
-                bucket = storage.data.get("bucket_path", "bucket")
+                bucket = storage.data.get(
+			        "bucket_path",
+			        storage.data.get(
+				        "bucket"
+			        )
+		        )
+                if not bucket:
+                    raise RuntimeError(
+                        f"Unable to parse bucket from storage block: {storage}"
+                    )
                 deployment_path = deployment.path.rstrip("/")
 
                 package_path = f"{bucket}/{deployment_path}/{module_name}"

diff --git a/pyproject.toml b/pyproject.toml
@@ -3,7 +3,7 @@ name = "block-cascade"
 packages = [
     {include = "block_cascade"}
 ]
-version = "2.5.6"
+version = "2.5.7"
 description = "Library for model training in multi-cloud environment."
 readme = "README.md"
 authors = ["Block"]