ensure_workspace

CHANGELOG.md

@@ -12,6 +12,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Fixed
 
 - Fixed bug with `Config.save()`.
+- Fixed bug with `create_image()` where it would fail if workspace doesn't exist yet.
 
 ## [v0.2.0](https://github.com/allenai/beaker-py/releases/tag/v0.2.0) - 2021-12-07
 

beaker/client.py

@@ -105,6 +105,16 @@ def get_workspace(self, workspace: Optional[str] = None) -> Dict[str, Any]:
             exceptions_for_status={404: WorkspaceNotFound(workspace_name)},
         ).json()
 
+    def ensure_workspace(self, workspace: str):
+        """
+        Ensure the given workspace exists.
+        """
+        try:
+            self.get_workspace(workspace)
+        except WorkspaceNotFound:
+            org, name = workspace.split("/")
+            self.request("workspaces", method="POST", data={"name": name, "org": org})
+
     def get_experiment(self, exp_id: str) -> Dict[str, Any]:
         """
         Get info about an experiment.
@@ -134,6 +144,8 @@ def create_experiment(
         workspace_name = workspace or self.config.default_workspace
         if workspace_name is None:
             raise ValueError("'workspace' argument required")
+        # Ensure workspace exists.
+        self.ensure_workspace(workspace_name)
         return self.request(
             f"workspaces/{urllib.parse.quote(workspace_name, safe='')}/experiments",
             method="POST",
@@ -229,6 +241,11 @@ def create_image(
 
         """
         workspace = workspace or self.config.default_workspace
+        if workspace is None:
+            raise ValueError("'workspace' argument required")
+
+        # Ensure workspace exists.
+        self.ensure_workspace(workspace)
 
         # Get local Docker image object.
         image = self.docker.images.get(image_tag)