fahmiaziz98 commited on
Commit
d089521
·
1 Parent(s): 7312439

frist commit

Browse files
Files changed (1) hide show
  1. scripts/s3.py +1 -1
scripts/s3.py CHANGED
@@ -32,7 +32,7 @@ def download_model_from_s3(local_path: Path, s3_prefix: str):
32
  os.makedirs(os.path.dirname(local_file), exist_ok=True)
33
  logger.info(f"Downloading {s3_key} to {local_file}")
34
  s3.download_file(BUCKET_NAME, s3_key, local_file)
35
- logger.info(f"Downloaded {s3_key} to {local_file}")
36
  except Exception as e:
37
  logger(f"Failed to download model from S3: {e}")
38
  raise RuntimeError(f"Error downloading model from S3: {e}")
 
32
  os.makedirs(os.path.dirname(local_file), exist_ok=True)
33
  logger.info(f"Downloading {s3_key} to {local_file}")
34
  s3.download_file(BUCKET_NAME, s3_key, local_file)
35
+ logger.info(f"Completed download {s3_key} to {local_file}")
36
  except Exception as e:
37
  logger(f"Failed to download model from S3: {e}")
38
  raise RuntimeError(f"Error downloading model from S3: {e}")