Skip to content
Snippets Groups Projects
Commit fbf798be authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

fix

parent 492b3f16
No related merge requests found
......@@ -137,7 +137,7 @@ do
proteome_peptides.$subset.csv.bz2 \
--result-dtype "float16" \
--predictor netmhcpan4-$kind \
--chunk-size 10000 \
--chunk-size 5000 \
--allele $(cat alleles.txt) \
--out "$OUT_DIR" \
--worker-log-dir "$SCRATCH_DIR/$DOWNLOAD_NAME" \
......
......@@ -311,8 +311,6 @@ def run(argv=sys.argv[1:]):
for allele in alleles:
allele_to_chunk_index_to_predictions[allele] = {}
last_write_time_per_column = dict((col, 0.0) for col in result_df.columns)
def write_col(col):
out_path = os.path.join(
args.out, col_to_filename[col])
......@@ -322,6 +320,13 @@ def run(argv=sys.argv[1:]):
result_df[col].isnull().mean() * 100.0),
out_path)
print("Writing all columns.")
last_write_time_per_column = {}
for col in result_df.columns:
write_col(col)
last_write_time_per_column[col] = time.time()
print("Done writing all columns. Reading results.")
for worker_results in tqdm.tqdm(results, total=len(work_items)):
for (work_item_num, col_to_predictions) in worker_results:
for (col, predictions) in col_to_predictions.items():
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment