chore: fix lint issues

pull/14262/head
Bharat Ramanathan 1 year ago
parent a2f8a49d5d
commit ea6a6bb753

@ -42,18 +42,12 @@ class WeaveDataTrace(BaseTraceInstance):
# Login with API key first # Login with API key first
login_status = wandb.login(key=self.weave_api_key, verify=True, relogin=True) login_status = wandb.login(key=self.weave_api_key, verify=True, relogin=True)
if not login_status: if not login_status:
logger.error( logger.error("Failed to login to Weights & Biases with the provided API key")
"Failed to login to Weights & Biases with the provided API key"
)
raise ValueError("Weave login failed") raise ValueError("Weave login failed")
# Then initialize weave client # Then initialize weave client
self.weave_client = weave.init( self.weave_client = weave.init(
project_name=( project_name=(f"{self.entity}/{self.project_name}" if self.entity else self.project_name)
f"{self.entity}/{self.project_name}"
if self.entity
else self.project_name
)
) )
self.file_base_url = os.getenv("FILES_URL", "http://127.0.0.1:5001") self.file_base_url = os.getenv("FILES_URL", "http://127.0.0.1:5001")
self.calls: dict[str, Any] = {} self.calls: dict[str, Any] = {}

Loading…
Cancel
Save