badaoui HF Staff commited on
Commit
b145057
·
1 Parent(s): 652d3bf

code optimization :)

Browse files
Files changed (1) hide show
  1. time_series_gradio.py +10 -10
time_series_gradio.py CHANGED
@@ -161,7 +161,7 @@ def create_time_series_summary_gradio(historical_df: pd.DataFrame) -> dict:
161
 
162
  def create_model_time_series_gradio(historical_df: pd.DataFrame, model_name: str) -> dict:
163
  def empty_figs():
164
- ef = lambda p: go.Figure().update_layout(title=f"{model_name.upper()} - {p} Results Over Time",
165
  height=500, font=dict(size=16, color='#CCCCCC'), paper_bgcolor='#000000',
166
  plot_bgcolor='#1a1a1a', margin=dict(b=130)) or go.Figure()
167
  return {'amd_plot': ef('AMD'), 'nvidia_plot': ef('NVIDIA')}
@@ -182,23 +182,23 @@ def create_model_time_series_gradio(historical_df: pd.DataFrame, model_name: str
182
  if dd.empty:
183
  continue
184
  r = dd.iloc[0]
185
- p = r.get(f'success_{platform}', 0)
186
- f = r.get(f'failed_multi_no_{platform}', 0) + r.get(f'failed_single_no_{platform}', 0)
187
- s = r.get(f'skipped_{platform}', 0)
188
 
189
  pc = fc = sc = 0
190
  if i > 0:
191
- pd = md[md['date'] == dates[i-1]]
192
- if not pd.empty:
193
- pr = pd.iloc[0]
194
  pc = pr.get(f'success_{platform}', 0)
195
  fc = pr.get(f'failed_multi_no_{platform}', 0) + pr.get(f'failed_single_no_{platform}', 0)
196
  sc = pr.get(f'skipped_{platform}', 0)
197
 
198
  data.extend([
199
- {'date': date, 'count': p, 'test_type': 'Passed', 'change': p - pc},
200
- {'date': date, 'count': f, 'test_type': 'Failed', 'change': f - fc},
201
- {'date': date, 'count': s, 'test_type': 'Skipped', 'change': s - sc}
202
  ])
203
  return pd.DataFrame(data)
204
 
 
161
 
162
  def create_model_time_series_gradio(historical_df: pd.DataFrame, model_name: str) -> dict:
163
  def empty_figs():
164
+ ef = lambda plat: go.Figure().update_layout(title=f"{model_name.upper()} - {plat} Results Over Time",
165
  height=500, font=dict(size=16, color='#CCCCCC'), paper_bgcolor='#000000',
166
  plot_bgcolor='#1a1a1a', margin=dict(b=130)) or go.Figure()
167
  return {'amd_plot': ef('AMD'), 'nvidia_plot': ef('NVIDIA')}
 
182
  if dd.empty:
183
  continue
184
  r = dd.iloc[0]
185
+ passed = r.get(f'success_{platform}', 0)
186
+ failed = r.get(f'failed_multi_no_{platform}', 0) + r.get(f'failed_single_no_{platform}', 0)
187
+ skipped = r.get(f'skipped_{platform}', 0)
188
 
189
  pc = fc = sc = 0
190
  if i > 0:
191
+ prev_dd = md[md['date'] == dates[i-1]]
192
+ if not prev_dd.empty:
193
+ pr = prev_dd.iloc[0]
194
  pc = pr.get(f'success_{platform}', 0)
195
  fc = pr.get(f'failed_multi_no_{platform}', 0) + pr.get(f'failed_single_no_{platform}', 0)
196
  sc = pr.get(f'skipped_{platform}', 0)
197
 
198
  data.extend([
199
+ {'date': date, 'count': passed, 'test_type': 'Passed', 'change': passed - pc},
200
+ {'date': date, 'count': failed, 'test_type': 'Failed', 'change': failed - fc},
201
+ {'date': date, 'count': skipped, 'test_type': 'Skipped', 'change': skipped - sc}
202
  ])
203
  return pd.DataFrame(data)
204