Add wandb api key

opendashboards/assets/io.py

@@ -12,7 +12,7 @@ print(f'BASE_DIR = {BASE_DIR}')
 def load_runs(project, filters, min_steps=10):
     runs = []
     msg = st.empty()
-    for run in utils.get_runs(project, filters):
+    for run in utils.get_runs(project, filters, api_key=st.secrets['WANDB_API_KEY']):
         step = run.summary.get('_step',0)
         if step < min_steps:
             msg.warning(f'Skipped run `{run.name}` because it contains {step} events (<{min_steps})')

opendashboards/utils/utils.py

@@ -24,7 +24,7 @@ from pandas.api.types import is_list_like
 from typing import List, Dict, Any, Union
 
 
-def get_runs(project: str = "openvalidators", filters: Dict[str, Any] = None, return_paths: bool = False) -> List:
+def get_runs(project: str = "openvalidators", filters: Dict[str, Any] = None, return_paths: bool = False, api_key: str = None) -> List:
     """Download runs from wandb.
 
     Args:
@@ -35,7 +35,7 @@ def get_runs(project: str = "openvalidators", filters: Dict[str, Any] = None, re
     Returns:
         List[wandb.apis.public.Run]: List of runs or run paths (List[str]).
     """
-    api = wandb.Api()
+    api = wandb.Api(api_key=api_key)
     wandb.login(anonymous="allow")
 
     runs = api.runs(project, filters=filters)
@@ -45,7 +45,7 @@ def get_runs(project: str = "openvalidators", filters: Dict[str, Any] = None, re
         return runs
 
 
-def download_data(run_path: Union[str, List] = None, timeout: float = 600) -> pd.DataFrame:
+def download_data(run_path: Union[str, List] = None, timeout: float = 600, api_key: str = None) -> pd.DataFrame:
     """Download data from wandb.
 
     Args:
@@ -55,7 +55,7 @@ def download_data(run_path: Union[str, List] = None, timeout: float = 600) -> pd
     Returns:
         pd.DataFrame: Dataframe of event log.
     """
-    api = wandb.Api(timeout=timeout)
+    api = wandb.Api(api_key=api_key, timeout=timeout)
     wandb.login(anonymous="allow")
 
     if isinstance(run_path, str):