Spaces:
Running on Zero
Running on Zero
time rfd3 execution
Browse files- utils/pipelines.py +3 -0
utils/pipelines.py
CHANGED
|
@@ -9,6 +9,7 @@ import gzip
|
|
| 9 |
from utils.handle_files import *
|
| 10 |
import sys
|
| 11 |
import shutil
|
|
|
|
| 12 |
|
| 13 |
@spaces.GPU(duration=240)
|
| 14 |
def test_rfd3_from_notebook():
|
|
@@ -140,7 +141,9 @@ def generation_with_input_config(input_file, pdb_file, num_batches, num_designs_
|
|
| 140 |
if extra_args:
|
| 141 |
command += f" {extra_args}"
|
| 142 |
print(f"Running command: {command}")
|
|
|
|
| 143 |
subprocess.run(command, shell=True, check=True, capture_output=True, text=True)
|
|
|
|
| 144 |
|
| 145 |
|
| 146 |
results = []
|
|
|
|
| 9 |
from utils.handle_files import *
|
| 10 |
import sys
|
| 11 |
import shutil
|
| 12 |
+
from time import perf_counter
|
| 13 |
|
| 14 |
@spaces.GPU(duration=240)
|
| 15 |
def test_rfd3_from_notebook():
|
|
|
|
| 141 |
if extra_args:
|
| 142 |
command += f" {extra_args}"
|
| 143 |
print(f"Running command: {command}")
|
| 144 |
+
start = perf_counter()
|
| 145 |
subprocess.run(command, shell=True, check=True, capture_output=True, text=True)
|
| 146 |
+
print("Command took", perf_counter() - start, "seconds to run.")
|
| 147 |
|
| 148 |
|
| 149 |
results = []
|