Skip to content

Commit

Permalink
[deps] fix: make wandb optional dependency
Browse files Browse the repository at this point in the history
  • Loading branch information
eric-haibin-lin committed Dec 2, 2024
1 parent f969c6a commit af7fa67
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion verl/utils/tracking.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@
A unified tracking interface that supports logging data to different backend
"""

import wandb

from typing import List, Union

Expand All @@ -32,6 +31,7 @@ def __init__(self, project_name, experiment_name, default_backend: Union[str, Li
self.logger = {}

if 'tracking' in default_backend:
import wandb
wandb.init(project=project_name, name=experiment_name, config=config)
self.logger['tracking'] = wandb

Expand Down

0 comments on commit af7fa67

Please sign in to comment.