kashif HF staff commited on
Commit
188d8b2
1 Parent(s): 84a0e08

update plotly and show only one legend for mean

Browse files
Files changed (2) hide show
  1. make_plot.py +3 -1
  2. requirements.txt +1 -1
make_plot.py CHANGED
@@ -107,6 +107,8 @@ def plot_forecast(df: pd.DataFrame, forecasts: List[pd.DataFrame]):
107
  mode="lines",
108
  name="Mean Forecast",
109
  line=dict(color="red", dash="dash"),
 
 
110
  )
111
  )
112
 
@@ -115,7 +117,7 @@ def plot_forecast(df: pd.DataFrame, forecasts: List[pd.DataFrame]):
115
  title=f"{df.columns[0]} Forecast",
116
  yaxis=dict(title=df.columns[0]),
117
  showlegend=True,
118
- legend=dict(x=0, y=1, font=dict(size=16)),
119
  hovermode="x", # Enable x-axis hover for better interactivity
120
  )
121
 
 
107
  mode="lines",
108
  name="Mean Forecast",
109
  line=dict(color="red", dash="dash"),
110
+ legendgroup="mean forecast",
111
+ showlegend=i == 0,
112
  )
113
  )
114
 
 
117
  title=f"{df.columns[0]} Forecast",
118
  yaxis=dict(title=df.columns[0]),
119
  showlegend=True,
120
+ legend=dict(x=0, y=1),
121
  hovermode="x", # Enable x-axis hover for better interactivity
122
  )
123
 
requirements.txt CHANGED
@@ -1,3 +1,3 @@
1
  gluonts[torch,pro]
2
  pandas
3
- plotly
 
1
  gluonts[torch,pro]
2
  pandas
3
+ plotly>=5.15.0