Skip to content

Fix 1130 and 1102 #1154

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

Jump to bottom
Merged
jlarson4 merged 11 commits into from
Jan 20, 2026
Merged

Fix 1130 and 1102 #1154

jlarson4 merged 11 commits into from
Jan 20, 2026
+40 −10

Conversation

@kapedalex
Copy link
Contributor

@kapedalex kapedalex commented Jan 8, 2026 •
edited
Loading

Description

Moved wandb import and initialization inside the train function to make it an optional dependency. Added is_library_available check to prevent crashes if wandb is not installed and for future reference.

Fixes #1130 and #1102

Type of change

  • Bug fix

Checklist:

  • I have commented my code, particularly in hard-to-understand areas
  • I have made corresponding changes to the documentation
  • My changes generate no new warnings
  • I have added tests that prove my fix is effective or that my feature works
  • New and existing unit tests pass locally with my changes
  • I have not rewritten tests relating to key interfaces which would affect backward compatibility

@kapedalex
Copy link
Contributor Author

kapedalex commented Jan 8, 2026

As I can see there are only 'No space left on device' errors, so tests passed.
To fix them we could use this: https://github.com/jlumbroso/free-disk-space, but I worry it might break home runner. I will leave link here as potential solution for later.

@kapedalex kapedalex changed the title Fix 1130 Fix 1130 and 1102 Jan 9, 2026
@jlarson4 jlarson4 merged commit edef345 into TransformerLensOrg:dev Jan 20, 2026
36 of 39 checks passed
jlarson4 added a commit that referenced this pull request Feb 3, 2026
@jlarson4
Fix WandB dependency and add is_library_available (#1154)
ad25921 
Cherry-picked from v2.17.0 commit edef345
Adapted for dev-3.x utilities structure

- Moved wandb import inside if config.wandb block
- Added is_library_available check before importing wandb
- Created utilities/library_utils.py with is_library_available function
- Exported is_library_available from utilities/__init__.py and utils.py
- Updated train.py to use library_utils.is_library_available
- Added test for is_library_available
- Removed poetry.lock (dev-3.x uses uv)

Original commit: edef345 Fix 1130 and 1102 (#1154)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

3 participants

@kapedalex @jlarson4 @bryce13950