gabboud commited on
Commit
833a14a
·
1 Parent(s): b8f3849

fix extraction of deisgn number

Browse files
Files changed (1) hide show
  1. utils/pipelines.py +1 -0
utils/pipelines.py CHANGED
@@ -132,6 +132,7 @@ def unconditional_generation_with_input_config(input_file, num_batches, num_desi
132
  results = []
133
  for file_name in os.listdir(directory):
134
  if file_name.endswith(".cif.gz"):
 
135
  terms = file_name.split("_")
136
  model_index = terms.index("model")
137
  batch = int(terms[model_index - 1])
 
132
  results = []
133
  for file_name in os.listdir(directory):
134
  if file_name.endswith(".cif.gz"):
135
+ name = os.path.basename(file_name).split(".")[0] #filename without extension
136
  terms = file_name.split("_")
137
  model_index = terms.index("model")
138
  batch = int(terms[model_index - 1])