dustalov commited on
Commit
240edb2
1 Parent(s): 994d3c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -15,7 +15,7 @@
15
  __author__ = 'Dmitry Ustalov'
16
  __license__ = 'Apache 2.0'
17
 
18
- import typing
19
 
20
  import gradio as gr
21
  import numpy as np
@@ -27,11 +27,8 @@ from plotly.graph_objects import Figure
27
 
28
  def visualize(df_pairwise: pd.DataFrame) -> Figure:
29
  fig = px.imshow(df_pairwise, color_continuous_scale='RdBu', text_auto='.2f')
30
-
31
  fig.update_layout(xaxis_title='Loser', yaxis_title='Winner', xaxis_side='top')
32
-
33
  fig.update_traces(hovertemplate='Winner: %{y}<br>Loser: %{x}<br>Fraction of Wins: %{z}<extra></extra>')
34
-
35
  return fig
36
 
37
 
@@ -85,7 +82,7 @@ def aggregate(wins: npt.NDArray[np.int64], ties: npt.NDArray[np.int64],
85
  return pi
86
 
87
 
88
- def handler(file: typing.IO[bytes], seed: int) -> typing.Tuple[pd.DataFrame, Figure]:
89
  if file is None:
90
  raise gr.Error('File must be uploaded')
91
 
 
15
  __author__ = 'Dmitry Ustalov'
16
  __license__ = 'Apache 2.0'
17
 
18
+ from typing import IO, Tuple
19
 
20
  import gradio as gr
21
  import numpy as np
 
27
 
28
  def visualize(df_pairwise: pd.DataFrame) -> Figure:
29
  fig = px.imshow(df_pairwise, color_continuous_scale='RdBu', text_auto='.2f')
 
30
  fig.update_layout(xaxis_title='Loser', yaxis_title='Winner', xaxis_side='top')
 
31
  fig.update_traces(hovertemplate='Winner: %{y}<br>Loser: %{x}<br>Fraction of Wins: %{z}<extra></extra>')
 
32
  return fig
33
 
34
 
 
82
  return pi
83
 
84
 
85
+ def handler(file: IO[bytes], seed: int) -> Tuple[pd.DataFrame, Figure]:
86
  if file is None:
87
  raise gr.Error('File must be uploaded')
88