Skip to content

Commit bf3c2da

Browse files
committed
fix
Signed-off-by: Hemil Desai <[email protected]>
1 parent e1d8727 commit bf3c2da

File tree

1 file changed

+12
-7
lines changed
  • src/nemo_run/core/packaging

1 file changed

+12
-7
lines changed

src/nemo_run/core/packaging/git.py

Lines changed: 12 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -161,13 +161,18 @@ def package(self, path: Path, job_dir: str, name: str) -> str:
161161
ctx.run(include_pattern_cmd)
162162

163163
with ctx.cd(git_base_path):
164-
# Extract and repack approach that works on all platforms
165-
temp_dir = f"temp_extract_{pattern_file_id}"
166-
ctx.run(f"mkdir -p {temp_dir}")
167-
ctx.run(f"tar xf {output_file}.tmp -C {temp_dir}")
168-
ctx.run(f"tar xf {pattern_tar_file_name} -C {temp_dir}")
169-
ctx.run(f"tar cf {output_file}.tmp -C {temp_dir} .")
170-
ctx.run(f"rm -rf {temp_dir} {pattern_tar_file_name}")
164+
if os.uname().sysname == 'Linux':
165+
# On Linux, directly concatenate tar files
166+
ctx.run(f"tar Af {output_file}.tmp {pattern_tar_file_name}")
167+
ctx.run(f"rm {pattern_tar_file_name}")
168+
else:
169+
# Extract and repack approach for other platforms
170+
temp_dir = f"temp_extract_{pattern_file_id}"
171+
ctx.run(f"mkdir -p {temp_dir}")
172+
ctx.run(f"tar xf {output_file}.tmp -C {temp_dir}")
173+
ctx.run(f"tar xf {pattern_tar_file_name} -C {temp_dir}")
174+
ctx.run(f"tar cf {output_file}.tmp -C {temp_dir} .")
175+
ctx.run(f"rm -rf {temp_dir} {pattern_tar_file_name}")
171176

172177
gzip_cmd = f"gzip -c {output_file}.tmp > {output_file}"
173178
rm_cmd = f"rm {output_file}.tmp"

0 commit comments

Comments
 (0)