gabboud commited on
Commit
8705e46
·
1 Parent(s): 464a533
Files changed (1) hide show
  1. utils/pipelines.py +1 -1
utils/pipelines.py CHANGED
@@ -71,7 +71,7 @@ def unconditional_generation(num_batches, num_designs_per_batch, length):
71
  results = []
72
  for batch in range(num_batches):
73
  for design in range(num_designs_per_batch):
74
- file_name = os.path.join(directory, f"_{batch}_{design}.cif.gz")
75
  results.append({"batch": batch, "design": design, "file": file_name, "pdb": cif_gz_to_pdb(file_name)})
76
 
77
  print(results)
 
71
  results = []
72
  for batch in range(num_batches):
73
  for design in range(num_designs_per_batch):
74
+ file_name = os.path.join(directory, f"_{batch}_model_{design}.cif.gz")
75
  results.append({"batch": batch, "design": design, "file": file_name, "pdb": cif_gz_to_pdb(file_name)})
76
 
77
  print(results)