mirror of
https://github.com/webclinic017/drift.git
synced 2026-07-27 18:57:55 +00:00
Mark Aron Szulyovszky
19 lines
329 B
Python
19 lines
329 B
Python
import wandb
|
|
import os
|
|
|
|
|
|
def get_wandb():
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
""" 0. Login to Weights and Biases """
|
|
wsb_token = os.environ.get("WANDB_API_KEY")
|
|
if wsb_token:
|
|
wandb.login(key=wsb_token)
|
|
return wandb
|
|
else:
|
|
return None # wandb.login()
|
|
|
|
# return wandb
|