Skip to content

Commit 1092d74

Browse files
committed
Update print message in check_existing_images
1 parent 385b12f commit 1092d74

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

src/cautiousrobot/__main__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ def process_checksums(img_dir, metadata_path, args, source_df):
102102

103103
# Verify numbers
104104
checksum_df = pd.read_csv(checksum_path, low_memory=False)
105-
expected_num_imgs = source_df.shape[0] - (args.starting_idx)
105+
expected_num_imgs = source_df.shape[0]
106106
print(f"There are {checksum_df.shape[0]} files in {img_dir}. Based on {args.input_file}, there should be {expected_num_imgs} images.")
107107

108108
return checksum_df, expected_num_imgs

src/cautiousrobot/utils.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,6 @@ def check_existing_images(csv_path, img_dir, source_df, filename_col, subfolders
158158
else:
159159
# Print directory status message - pre-download
160160
num_existing = len(existing_files)
161-
expected_num = df.shape[0] - (starting_idx)
162-
print(f"There are {num_existing} files in {img_dir}. Based on {csv_path}, there should be {expected_num} images.")
161+
print(f"There are {num_existing} of desired files in {img_dir}. Based on {csv_path}, {filtered_df.shape[0]} images should be downloaded.")
163162

164163
return df, filtered_df

0 commit comments

Comments
 (0)