fix plot when there are more than 3 windows
Browse files- make_plot.py +1 -1
make_plot.py
CHANGED
|
@@ -84,7 +84,7 @@ def plot_forecast(df: pd.DataFrame, forecasts: List[pd.DataFrame]):
|
|
| 84 |
# Add the forecast traces
|
| 85 |
colors = ["green", "blue", "purple"]
|
| 86 |
for i, forecast in enumerate(forecasts):
|
| 87 |
-
color = colors[i]
|
| 88 |
for sample in forecast.samples:
|
| 89 |
fig.add_trace(
|
| 90 |
go.Scatter(
|
|
|
|
| 84 |
# Add the forecast traces
|
| 85 |
colors = ["green", "blue", "purple"]
|
| 86 |
for i, forecast in enumerate(forecasts):
|
| 87 |
+
color = colors[i % len(colors)]
|
| 88 |
for sample in forecast.samples:
|
| 89 |
fig.add_trace(
|
| 90 |
go.Scatter(
|