Skip to content

Commit

Permalink
fix(sdk): fix probing static IPU info (#4464)
Browse files Browse the repository at this point in the history
  • Loading branch information
dmitryduev authored and andrewtruong committed Dec 2, 2022
1 parent b5e31ad commit 4aa7f32
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions wandb/sdk/internal/system/assets/ipu.py
Expand Up @@ -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",
}
)

Expand Down

0 comments on commit 4aa7f32

Please sign in to comment.