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_status = wandb.login(key=self.weave_api_key, verify=True, relogin=True)
if not login_status:
logger.error(
"Failed to login to Weights & Biases with the provided API key"
)
logger.error("Failed to login to Weights & Biases with the provided API key")
raise ValueError("Weave login failed")
# Then initialize weave client
self.weave_client = weave.init(
project_name=(
f"{self.entity}/{self.project_name}"
if self.entity
else self.project_name
)
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.calls: dict[str, Any] = {}

Loading…
Cancel
Save