edmundmiller commited on
Commit
0d164eb
·
unverified ·
1 Parent(s): 1f3e2b4

fix: out_dir => outdir

Browse files
Files changed (1) hide show
  1. predict_chromosome.py +5 -5
predict_chromosome.py CHANGED
@@ -216,7 +216,7 @@ def predict_and_write(
216
  start_time = time.time()
217
 
218
  denoised_anchor_to_anchor.to_csv(
219
- os.path.join(out_dir, chromosome + ".denoised.anchor.to.anchor"),
220
  sep="\t",
221
  index=False,
222
  header=False,
@@ -242,7 +242,7 @@ if __name__ == "__main__":
242
  help="path to model architecture .json file (by default it is assumed to be the same as the weights file)",
243
  )
244
  parser.add_argument(
245
- "--out_dir",
246
  type=str,
247
  help="directory where the output interaction file will be stored",
248
  )
@@ -299,7 +299,7 @@ if __name__ == "__main__":
299
  json_file = args.json_file
300
  else:
301
  json_file = args.h5_file.replace("h5", "json")
302
- out_dir = args.out_dir
303
  anchor_dir = args.anchor_dir
304
  chromosome = args.chromosome
305
  small_matrix_size = args.small_matrix_size
@@ -309,7 +309,7 @@ if __name__ == "__main__":
309
  val_cols = args.val_cols
310
  keep_zeros = args.keep_zeros
311
 
312
- os.makedirs(out_dir, exist_ok=True)
313
 
314
  with open(json_file, "r") as f:
315
  model = model_from_json(f.read()) # load model
@@ -318,7 +318,7 @@ if __name__ == "__main__":
318
  model,
319
  full_matrix_dir,
320
  input_name,
321
- out_dir,
322
  anchor_dir,
323
  chromosome,
324
  small_matrix_size,
 
216
  start_time = time.time()
217
 
218
  denoised_anchor_to_anchor.to_csv(
219
+ os.path.join(outdir, chromosome + ".denoised.anchor.to.anchor"),
220
  sep="\t",
221
  index=False,
222
  header=False,
 
242
  help="path to model architecture .json file (by default it is assumed to be the same as the weights file)",
243
  )
244
  parser.add_argument(
245
+ "--outdir",
246
  type=str,
247
  help="directory where the output interaction file will be stored",
248
  )
 
299
  json_file = args.json_file
300
  else:
301
  json_file = args.h5_file.replace("h5", "json")
302
+ outdir = args.outdir
303
  anchor_dir = args.anchor_dir
304
  chromosome = args.chromosome
305
  small_matrix_size = args.small_matrix_size
 
309
  val_cols = args.val_cols
310
  keep_zeros = args.keep_zeros
311
 
312
+ os.makedirs(outdir, exist_ok=True)
313
 
314
  with open(json_file, "r") as f:
315
  model = model_from_json(f.read()) # load model
 
318
  model,
319
  full_matrix_dir,
320
  input_name,
321
+ outdir,
322
  anchor_dir,
323
  chromosome,
324
  small_matrix_size,