gavindiaz/drift
1
0
Fork 0
mirror of https://github.com/webclinic017/drift.git synced 2026-08-18 05:18:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(Linter): reformatted code with black (#211)

Browse Source 
* chore(Linter): reformatted code with black

* Create black.yaml
This commit is contained in:
Mark Aron Szulyovszky
2022-02-17 19:22:17 +01:00
committed by GitHub
parent f3fee4a4e1
commit 8dd2d88740
101 changed files with 2595 additions and 2319 deletions
Download Patch File Download Diff File Expand all files Collapse all files
reporting/wandb.py
+24 -16
View File
@@ -4,41 +4,49 @@ from typing import Optional
from utils.helpers import weighted_average
from training.types import Stats
def launch_wandb(project_name:str, default_config: RawConfig, sweep:bool=False) -> Optional[object]:
def launch_wandb(
project_name: str, default_config: RawConfig, sweep: bool = False
) -> Optional[object]:
from wandb_setup import get_wandb
wandb = get_wandb()
wandb = get_wandb()
if wandb is None:
raise Exception("Wandb can not be initalized, the environment variable WANDB_API_KEY is missing (can also use .env file)")
raise Exception(
"Wandb can not be initalized, the environment variable WANDB_API_KEY is missing (can also use .env file)"
)
elif sweep:
wandb.init(project=project_name, config = vars(default_config))
wandb.init(project=project_name, config=vars(default_config))
return wandb
else:
wandb.init(project=project_name, config = vars(default_config), reinit=True)
wandb.init(project=project_name, config=vars(default_config), reinit=True)
return wandb
def override_config_with_wandb_values(wandb: Optional[object], raw_config: RawConfig) -> RawConfig:
if wandb is None: return raw_config
def override_config_with_wandb_values(
wandb: Optional[object], raw_config: RawConfig
) -> RawConfig:
if wandb is None:
return raw_config
wandb_config: dict = wandb.config
config_dict = vars(raw_config)
for k in config_dict:
config_dict[k] = wandb_config[k]
return RawConfig(**config_dict)
def send_report_to_wandb(stats: Stats, wandb:Optional[object]):
if wandb is None: return
def send_report_to_wandb(stats: Stats, wandb: Optional[object]):
if wandb is None:
return
run = wandb.run
run.save()
for key, value in stats.items():
run.log({ key: value })
run.log({key: value})
run.finish()