Spaces:
Running on Zero
Running on Zero
fix extraction of deisgn number
Browse files- 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])
|