Skip to content

Commit 4d7d3b0

Browse files
committed
fix: cleanup
1 parent c12623e commit 4d7d3b0

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

src/pipeline/pipeline.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -250,41 +250,41 @@ def download_s5cmd():
250250
# Check if s5cmd already exists
251251
if os.path.exists(S5CMD_PATH):
252252
return S5CMD_PATH
253-
253+
254254
s5cmd_url = "https://aws-gcr-solutions-us-east-1.s3.us-east-1.amazonaws.com/easy-model-deployer/pipeline/s5cmd.zip"
255-
255+
256256
try:
257257
# Download and extract
258258
zip_path = S5CMD_PATH + ".zip"
259259
urllib.request.urlretrieve(s5cmd_url, zip_path)
260-
260+
261261
with zipfile.ZipFile(zip_path, 'r') as zip_ref:
262262
zip_ref.extractall(".")
263-
263+
264264
os.remove(zip_path) # Clean up zip file
265-
265+
266266
# Make s5cmd executable
267267
if os.path.exists(S5CMD_PATH):
268268
os.chmod(S5CMD_PATH, os.stat(S5CMD_PATH).st_mode | 0o755)
269269
return S5CMD_PATH
270270
else:
271271
raise FileNotFoundError("Required component(s5cmd) installation failed")
272-
272+
273273
except Exception as e:
274274
raise RuntimeError("Required component(s5cmd) download failed")
275275

276276
if __name__ == "__main__":
277277
t0 = time.time()
278278
start_time = time.time()
279279
args = parse_args()
280-
280+
281281
# Download s5cmd
282282
download_s5cmd()
283283

284284
if not os.path.exists(S5CMD_PATH):
285285
logger.error("Required component(s5cmd) not found")
286286
sys.exit(1)
287-
287+
288288
os.chmod(S5CMD_PATH, os.stat(S5CMD_PATH).st_mode | 0o100)
289289
extra_params = args.extra_params
290290
for k,v in extra_params.items():

0 commit comments

Comments
 (0)