expr_tracker 0.1.4__tar.gz → 0.1.5__tar.gz

{expr_tracker-0.1.4 → expr_tracker-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: expr_tracker
-Version: 0.1.4
+Version: 0.1.5
 Summary: Add your description here
 Author-email: HSPK <whxway@whu.edu.cn>
 Requires-Python: >=3.10

{expr_tracker-0.1.4 → expr_tracker-0.1.5}/src/expr_tracker/tracker.py

@@ -1,6 +1,7 @@
 from typing import Literal
 from contextvars import ContextVar
 from loguru import logger
+import os
 
 _tracker: ContextVar["Tracker | None"] = ContextVar("tracker", default=None)
 
@@ -12,6 +13,10 @@ def get_backend(backend: str | object):
             try:
                 import wandb
 
+                wandb.login(
+                    key=os.getenv("WANDB_API_KEY", None),
+                    host=os.getenv("WANDB_HOST", None),
+                )
                 return wandb
             except ImportError:
                 raise ImportError("WandB backend is not installed.")