Spaces:
Running on Zero
Running on Zero
fix path
Browse files- 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}
|
| 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)
|