From 1a03e378828e7b06a7fb30e4fcdeaf303fcf3e5c Mon Sep 17 00:00:00 2001
From: MichaelClifford <mcliffor@redhat.com>
Date: Tue, 11 Apr 2023 15:13:49 -0400
Subject: [PATCH] make workspace user accessible

---
 src/codeflare_sdk/job/jobs.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/codeflare_sdk/job/jobs.py b/src/codeflare_sdk/job/jobs.py
index d89269220..6b5ce0a53 100644
--- a/src/codeflare_sdk/job/jobs.py
+++ b/src/codeflare_sdk/job/jobs.py
@@ -64,6 +64,7 @@ def __init__(
         rdzv_backend: str = None,
         scheduler_args: Optional[Dict[str, str]] = None,
         image: Optional[str] = None,
+        workspace: Optional[str] = f"file://{Path.cwd()}",
     ):
         if bool(script) == bool(m):  # logical XOR
             raise ValueError(
@@ -87,6 +88,7 @@ def __init__(
             scheduler_args if scheduler_args is not None else dict()
         )
         self.image = image
+        self.workspace = workspace
 
     def _dry_run(self, cluster: "Cluster"):
         j = f"{cluster.config.max_worker}x{max(cluster.config.gpu, 1)}"  # # of proc. = # of gpus
@@ -113,7 +115,7 @@ def _dry_run(self, cluster: "Cluster"):
             ),
             scheduler=cluster.torchx_scheduler,
             cfg=cluster.torchx_config(**self.scheduler_args),
-            workspace=f"file://{Path.cwd()}",
+            workspace=self.workspace,
         )
 
     def _missing_spec(self, spec: str):