Spaces:
Sleeping
Sleeping
File size: 3,368 Bytes
b693c5c c121aed f3262d9 5e35297 1af6422 363c4c6 bc55598 363d1a5 0fa040d 79e6cc7 17d61e7 fc7f7cb 3ebe0a0 17d61e7 b693c5c c121aed 3ebe0a0 17d61e7 f56cccc 17d61e7 5c74948 17d61e7 df8c933 17d61e7 e28c33a 3ebe0a0 14eb75b 3ebe0a0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 |
from github import Github
import os
import streamlit as st
import datetime
import plotly.figure_factory as ff
import plotly.graph_objects as go
import pandas as pd
import math
import copy
st.set_page_config(layout="wide")
name2repo = [("Training", "bigcode-project/Megatron-LM"),
("Dataset", "bigcode-project/bigcode-dataset"),
# ("Evaluation", "bigcode-project/bigcode-evaluation-harness"),
# ("Inference", "bigcode-project/bigcode-inference-benchmark"),
]
g = Github(os.environ.get('github'))
for name, repo_name in name2repo:
repo = g.get_repo(repo_name)
df = list()
all_status = list()
for milestone in repo.get_milestones():
desc = dict()
for line in milestone.description.split('\n'):
tmp = line.split(":")
if len(tmp) > 1:
desc[tmp[0].lower()] = tmp[1].lower().strip()
task_name = f"""<a href="https://www.github.com/{repo_name}/milestone/{milestone.number}", target="_black">{milestone.title}</a>"""
if desc['status'] not in all_status:
all_status.append(desc['status'])
df.append(dict(Task=task_name,
Start=desc['start date'],
Finish=milestone.due_on.strftime('%Y-%m-%d'),
Resource=desc['status'],
Description=desc['leader']))
colors = {'not started': 'rgb(217, 217, 217)',
'in progress': 'rgb(147, 196, 125)',
'high priority - on track': 'rgb(234, 153, 153)',
'high priority - help needed': 'rgb(255, 0, 0)',
'completed': 'rgb(56, 118, 29)'}
if len(all_status) == 0:
task_name = "None"
for key in colors.keys():
if key not in all_status:
df.append(dict(Task=task_name,
Start='2023-04-02',
Finish='2023-04-02',
Resource=key))
fig = ff.create_gantt(df, colors=colors,
index_col='Resource',
show_colorbar=True,
show_hover_fill=True,
group_tasks=True,
title="BigCode planning")
fig.update_xaxes(ticks= "outside",
ticklabelmode= "period",
tickformat="%b",
tickcolor= "black",
ticklen=10,
range=[datetime.datetime(2022, 12, 30),
datetime.datetime(2023, 4, 2)],
minor=dict(
ticklen=4,
dtick=7*24*60*60*1000,
tick0="2023-01-01",
griddash='dot',
gridcolor='white')
)
fig.update_layout(margin=go.layout.Margin(l=250))
fig.layout.xaxis.rangeselector = None # remove range selector on top
fig.add_vline(x=datetime.datetime.now().strftime('%Y-%m-%d'), line_width=3, line_dash="dash", line_color="black")
fig.add_annotation({
"x": datetime.datetime.now().strftime('%Y-%m-%d'),
"y": fig.layout.yaxis['range'][1],
"yshift": 10,
"text": "Today",
"align": "left",
"showarrow": False,
})
fig.add_hline(y=-0.5, line_width=1, line_color="grey")
fig.add_hline(y=2.5, line_width=1, line_color="grey")
st.plotly_chart(fig, use_container_width=True) |