Spaces:
Running on Zero
Running on Zero
typo
Browse files- utils/pipelines.py +2 -2
utils/pipelines.py
CHANGED
|
@@ -71,14 +71,14 @@ 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.
|
| 75 |
results.append({"batch": batch, "design": design, "file": file_name, "pdb": cif_gz_to_pdb(file_name)})
|
| 76 |
|
| 77 |
print(results)
|
| 78 |
return directory, results
|
| 79 |
|
| 80 |
except Exception as e:
|
| 81 |
-
|
| 82 |
|
| 83 |
def collect_outputs(gen_directory, num_batches, num_designs_per_batch):
|
| 84 |
try:
|
|
|
|
| 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)
|
| 78 |
return directory, results
|
| 79 |
|
| 80 |
except Exception as e:
|
| 81 |
+
raise RuntimeError(f"Error during generation: {str(e)}")
|
| 82 |
|
| 83 |
def collect_outputs(gen_directory, num_batches, num_designs_per_batch):
|
| 84 |
try:
|