Skip to content

Commit

Permalink
process-bound
Browse files Browse the repository at this point in the history
  • Loading branch information
davedavemckay committed Jan 31, 2025
1 parent f198832 commit a27be82
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions scripts/process_collated_zips.py
Original file line number Diff line number Diff line change
Expand Up @@ -244,14 +244,14 @@ def error(self, message):
keys_df[keys_df['is_zipfile'] == False]['contents'] = None

if list_zips:
print(keys_df[keys_df['is_zipfile'] == True]['key'].compute(scheduler='single-threaded'))
print(keys_df[keys_df['is_zipfile'] == True]['key'].compute(scheduler='processes'))

if extract:
print('Extracting zip files...')
keys_df['extract'] = keys_df.apply(verify_zip_contents, meta=('extract', 'bool'), keys_df=keys_df, axis=1)
keys_df['extracted and uploaded'] = keys_df.apply(extract_and_upload, conn=conn, bucket_name=bucket_name, meta=('extracted and uploaded', 'bool'), axis=1)
print('Zip files extracted and uploaded:')
print(keys_df[keys_df['extracted and uploaded'] == True]['key'].compute(scheduler='single-threaded'))
print(keys_df[keys_df['extracted and uploaded'] == True]['key'].compute(scheduler='processes'))

print('Done.')

Expand Down

0 comments on commit a27be82

Please sign in to comment.