fix: download result

CloudOS-Group3 · Jun 9, 2024 · 4f0ad71 · 4f0ad71
1 parent c80d3f3
commit 4f0ad71
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkg/gpu/image/gpu_server.py b/pkg/gpu/image/gpu_server.py
@@ -133,7 +133,8 @@ def send_back_result(status):
             sftp = ssh.open_sftp()
 
     # download the result file
-    sftp.get(f'{remote_dir}/{job_id}.out', f'./{job_name}.out')
+    sftp.get(f'{remote_dir}/{job_id}.out', f'{local_dir}/{job_name}.out')
+    sftp.get(f'{remote_dir}/{job_id}.err', f'{local_dir}/{job_name}.err')
     print(f"Downloaded result file {job_id}.out.")
     send_back_result('success')