We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 81accf3 + d7560b3 commit 2ba4c2dCopy full SHA for 2ba4c2d
job_executor/adapter/local_storage.py
@@ -428,9 +428,13 @@ def delete_archived_input(dataset_name: str):
428
def get_input_tar_size_in_bytes(dataset_name: str) -> int:
429
"""
430
Checks the size in bytes of the dataset.tar file.
431
- returns size in bytes
+ Returns size in bytes or 0 if the file does not exist.
432
433
tar_path = INPUT_DIR / f"{dataset_name}.tar"
434
+
435
if not tar_path.exists():
- return 0
436
- return os.path.getsize(tar_path)
+ tar_path = INPUT_DIR / "archive" / f"{dataset_name}.tar"
437
438
+ if tar_path.exists():
439
+ return os.path.getsize(tar_path)
440
+ return 0
0 commit comments