Exclude profiles of training admins
Browse files
dashboard_utils/bubbles.py
CHANGED
|
@@ -10,6 +10,7 @@ from requests_futures.sessions import FuturesSession
|
|
| 10 |
|
| 11 |
from dashboard_utils.time_tracker import _log, simple_time_tracker
|
| 12 |
|
|
|
|
| 13 |
URL_QUICKSEARCH = "https://huggingface.co/api/quicksearch?"
|
| 14 |
WANDB_REPO = st.secrets["WANDB_REPO_INDIVIDUAL_METRICS"]
|
| 15 |
CACHE_TTL = 100
|
|
@@ -73,8 +74,11 @@ def get_serialized_data_points():
|
|
| 73 |
serialized_data_points = {}
|
| 74 |
latest_timestamp = None
|
| 75 |
for run in runs:
|
| 76 |
-
run_summary = run.summary._json_dict
|
| 77 |
run_name = run.name
|
|
|
|
|
|
|
|
|
|
|
|
|
| 78 |
state = run.state
|
| 79 |
|
| 80 |
if run_name in serialized_data_points:
|
|
|
|
| 10 |
|
| 11 |
from dashboard_utils.time_tracker import _log, simple_time_tracker
|
| 12 |
|
| 13 |
+
EXCLUDED_PROFILES = {'borzunov', 'justheuristic', 'training-transformers-together-machine'}
|
| 14 |
URL_QUICKSEARCH = "https://huggingface.co/api/quicksearch?"
|
| 15 |
WANDB_REPO = st.secrets["WANDB_REPO_INDIVIDUAL_METRICS"]
|
| 16 |
CACHE_TTL = 100
|
|
|
|
| 74 |
serialized_data_points = {}
|
| 75 |
latest_timestamp = None
|
| 76 |
for run in runs:
|
|
|
|
| 77 |
run_name = run.name
|
| 78 |
+
if run_name in EXCLUDED_PROFILES:
|
| 79 |
+
continue
|
| 80 |
+
|
| 81 |
+
run_summary = run.summary._json_dict
|
| 82 |
state = run.state
|
| 83 |
|
| 84 |
if run_name in serialized_data_points:
|