Skip to content

Commit 1a03e37

Browse files
make workspace user accessible
1 parent c7e2bf1 commit 1a03e37

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/codeflare_sdk/job/jobs.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,7 @@ def __init__(
6464
rdzv_backend: str = None,
6565
scheduler_args: Optional[Dict[str, str]] = None,
6666
image: Optional[str] = None,
67+
workspace: Optional[str] = f"file://{Path.cwd()}",
6768
):
6869
if bool(script) == bool(m): # logical XOR
6970
raise ValueError(
@@ -87,6 +88,7 @@ def __init__(
8788
scheduler_args if scheduler_args is not None else dict()
8889
)
8990
self.image = image
91+
self.workspace = workspace
9092

9193
def _dry_run(self, cluster: "Cluster"):
9294
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"):
113115
),
114116
scheduler=cluster.torchx_scheduler,
115117
cfg=cluster.torchx_config(**self.scheduler_args),
116-
workspace=f"file://{Path.cwd()}",
118+
workspace=self.workspace,
117119
)
118120

119121
def _missing_spec(self, spec: str):

0 commit comments

Comments
 (0)