Skip to content

Commit 35e8818

Browse files
committed
Changed download and upload path
1 parent cf45c7d commit 35e8818

File tree

2 files changed

+4
-30
lines changed

2 files changed

+4
-30
lines changed

skills/rca-annotator/SKILL.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ cd skills/rca-annotator
5252
python scripts/cli.py download --job-id <job_id>
5353
```
5454

55-
Downloads from jumpbox `/tmp/analysis/<job_id>/` (or `/tmp/<job_id>`) to local `.analysis/<job_id>/`. If `JUMPBOX_URI` unset, validates local files only. Errors on missing remote directory, missing required files, or connection failure.
55+
Downloads from jumpbox `/usr/local/mlflow/<job_id>/` to local `.analysis/<job_id>/`. If `JUMPBOX_URI` unset, validates local files only. Errors on missing remote directory, missing required files, or connection failure.
5656

5757
---
5858

skills/rca-annotator/scripts/jumpbox_io.py

Lines changed: 3 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def download_from_jumpbox(job_id: str, jumpbox_uri: str | None = None) -> bool:
117117
print(f" Error: {e}")
118118
return False
119119

120-
remote_candidates = [f"/tmp/analysis/{job_id}", f"/tmp/{job_id}"]
120+
remote_candidates = [f"/usr/local/mlflow/{job_id}"]
121121
remote_dir = None
122122

123123
for candidate in remote_candidates:
@@ -185,10 +185,7 @@ def download_from_jumpbox(job_id: str, jumpbox_uri: str | None = None) -> bool:
185185

186186
def upload_to_jumpbox(job_id: str, jumpbox_uri: str | None = None) -> bool:
187187
"""
188-
Upload annotation_draft.json to jumpbox at /tmp/analysis/<job_id>/ or /tmp/<job_id>/.
189-
190-
Checks which remote directory exists and uploads there. Falls back to
191-
/tmp/analysis/<job_id>/ if neither exists yet.
188+
Upload annotation_draft.json to jumpbox at /usr/local/mlflow/<job_id>/.
192189
193190
Args:
194191
job_id: Job ID to upload annotation for
@@ -226,30 +223,7 @@ def upload_to_jumpbox(job_id: str, jumpbox_uri: str | None = None) -> bool:
226223
print(f" Annotation saved locally at: {local_file}")
227224
return False
228225

229-
remote_candidates = [f"/tmp/analysis/{job_id}", f"/tmp/{job_id}"]
230-
remote_dir = None
231-
232-
for candidate in remote_candidates:
233-
ssh_cmd = ["ssh"]
234-
if ssh_port:
235-
ssh_cmd.extend(["-p", ssh_port])
236-
ssh_cmd.extend([ssh_target, f"test -d {candidate}"])
237-
238-
try:
239-
subprocess.run(
240-
ssh_cmd,
241-
check=True,
242-
capture_output=True,
243-
timeout=30,
244-
)
245-
remote_dir = candidate
246-
break
247-
except (subprocess.CalledProcessError, subprocess.TimeoutExpired):
248-
continue
249-
250-
if remote_dir is None:
251-
remote_dir = remote_candidates[0]
252-
print(f" No existing remote directory found; defaulting to {remote_dir}")
226+
remote_dir = f"/usr/local/mlflow/{job_id}"
253227

254228
print(" Uploading annotation to jumpbox...")
255229
print(f" Local: {local_file}")

0 commit comments

Comments
 (0)