civerson916 commited on
Commit
84a5ab5
·
verified ·
1 Parent(s): 82f2d56

Added elapsed time

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -6,6 +6,7 @@ from opentelemetry import trace
6
  from evaluator import Evaluator
7
  from runner import Runner
8
  from settings import Settings
 
9
  import os
10
  import pandas as pd
11
  import gradio as gr
@@ -45,8 +46,13 @@ EMPTY_RESULTS_TABLE = pd.DataFrame(columns=['task_id', 'question', 'answer'])
45
  def run_one(profile: gr.OAuthProfile | None) -> pd.DataFrame:
46
  if not user_logged_in(profile):
47
  return LOGIN_MESSAGE, EMPTY_RESULTS_TABLE
 
48
  questions = [evaluator.get_one_question()]
49
- return "Complete.", runner.run_agent(questions, profile.username)
 
 
 
 
50
 
51
  def run_all(profile: gr.OAuthProfile | None) -> pd.DataFrame:
52
  if not user_logged_in(profile):
 
6
  from evaluator import Evaluator
7
  from runner import Runner
8
  from settings import Settings
9
+ import time
10
  import os
11
  import pandas as pd
12
  import gradio as gr
 
46
  def run_one(profile: gr.OAuthProfile | None) -> pd.DataFrame:
47
  if not user_logged_in(profile):
48
  return LOGIN_MESSAGE, EMPTY_RESULTS_TABLE
49
+ start_time = time.time()
50
  questions = [evaluator.get_one_question()]
51
+ question_answer_pairs = runner.run_agent(questions, profile.username)
52
+ end_time = time.time()
53
+ elapsed_time = end_time - start_time
54
+ message = f"Complete. Elapsed time: {elapsed_time} seconds"
55
+ return message, question_answer_pairs
56
 
57
  def run_all(profile: gr.OAuthProfile | None) -> pd.DataFrame:
58
  if not user_logged_in(profile):