Skip to content

fix wandb local converter #1596

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Zeyi-Lin merged 2 commits into from
May 6, 2026
Merged

fix wandb local converter #1596

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
b3c1a93
fix
Zeyi-Lin May 6, 2026
2ab3d18
fix
Zeyi-Lin May 6, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
46 changes: 24 additions & 22 deletions swanlab/converter/wb/wb_local_converter.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -409,7 +409,30 @@ def _pre_upload_cb(n):
continue
if key.startswith('_'):
continue
# Check if key has nested structure (e.g., "table/_type")
# Fast path: direct float conversion for scalars (all keys)
try:
scalar_dict[key] = float(value_json)
continue
except (ValueError, TypeError):
pass
# Slow path: full JSON parse for media objects (all keys)
try:
value = _json_loads(value_json)
if isinstance(value, int):
scalar_dict[key] = float(value)
continue
elif isinstance(value, dict) and "path" in value:
validated_path = self._validate_path(files_root_dir, value["path"])
if validated_path and os.path.exists(validated_path):
if value.get("_type") == "image-file":
media_dict[key] = swanlab.Image(validated_path)
continue
elif value.get("_type") == "audio-file":
media_dict[key] = swanlab.Audio(validated_path)
continue
except (ValueError, Exception):
pass
# Grouped path: keys with '/' that weren't handled above (tables, etc.)
if '/' in key:
base_key, sub_key = key.split('/', 1)
if base_key not in grouped_items:
Expand All @@ -418,27 +441,6 @@ def _pre_upload_cb(n):
grouped_items[base_key][sub_key] = _json_loads(value_json)
except (ValueError, Exception):
grouped_items[base_key][sub_key] = value_json
else:
# Fast path: direct float conversion for scalars
try:
scalar_dict[key] = float(value_json)
continue
except (ValueError, TypeError):
pass
# Slow path: full JSON parse
try:
value = _json_loads(value_json)
if isinstance(value, int):
scalar_dict[key] = float(value)
elif isinstance(value, dict) and "path" in value:
validated_path = self._validate_path(files_root_dir, value["path"])
if validated_path and os.path.exists(validated_path):
if value.get("_type") == "image-file":
media_dict[key] = swanlab.Image(validated_path)
elif value.get("_type") == "audio-file":
media_dict[key] = swanlab.Audio(validated_path)
except (ValueError, Exception):
pass

# Second pass: process grouped items for tables and media
for base_key, props in grouped_items.items():
Expand Down
Loading