gabboud commited on
Commit
2535c77
·
1 Parent(s): 4e40dbe

time rfd3 execution

Browse files
Files changed (1) hide show
  1. 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 = []