| """Use real wandb when installed; allow explicitly disabled training without it.""" | |
| try: | |
| import wandb as wandb | |
| except ModuleNotFoundError as exc: | |
| if exc.name != "wandb": | |
| raise | |
| class _DisabledWandb: | |
| def finish(*args, **kwargs): | |
| return None | |
| def __getattr__(self, name): | |
| raise RuntimeError( | |
| "wandb is not installed; launch with --disable-wandb" | |
| ) | |
| wandb = _DisabledWandb() | |