diff --git a/wandb/sdk/internal/system/assets/ipu.py b/wandb/sdk/internal/system/assets/ipu.py index 35b6ba24be4..123c0793da6 100644 --- a/wandb/sdk/internal/system/assets/ipu.py +++ b/wandb/sdk/internal/system/assets/ipu.py @@ -161,11 +161,12 @@ def probe(self) -> dict: device_count = len(device_data) devices = [] for i, device in enumerate(device_data): + device_metrics: Dict[str, str] = dict(device) devices.append( { - "id": device.get("id") or i, - "board ipu index": device.get("board ipu index"), - "board type": device.get("board type") or "unknown", + "id": device_metrics.get("id") or i, + "board ipu index": device_metrics.get("board ipu index"), + "board type": device_metrics.get("board type") or "unknown", } )