Skip to content

docs: fix broken GenerateCheckWeights notebook + repoint stale Colab link#18

Merged
LukasWodka merged 1 commit into
mainfrom
docs/fix-notebooks-and-colab-link
May 19, 2026
Merged

docs: fix broken GenerateCheckWeights notebook + repoint stale Colab link#18
LukasWodka merged 1 commit into
mainfrom
docs/fix-notebooks-and-colab-link

Conversation

@saadqbal

@saadqbal saadqbal commented May 19, 2026

Copy link
Copy Markdown
Collaborator

Summary

Splits the docs portion out of #15 (which was conflicting and stale). Clean cut against main, no CI workflows or merge noise.

  • Fix broken notebook on main. notebooks/GenerateCheckWeights.ipynb currently imports `from tracebloc import ModelWeights`, which doesn't exist in 0.8.x — running it raises ImportError. Rewritten using importlib + torch.save/pickle directly, matching the file layout the SDK expects (`.pth` for PyTorch, `.pkl` for TensorFlow).
  • Repoint stale Colab badge. README + training notebook both pointed at a Drive-hosted Colab copy (`drive/1N00idtp...`) that needed manual sync on every notebook change and had drifted. Switched to the `colab.research.google.com/github/...` form so the canonical notebook loads from `main` directly.
  • Minor UX in training guide. `training_object` → `training`, "Dataset" wording aligned with the use-case panel label, Drive-save reminder inlined in Prerequisites, `.pth`/`.pkl` extension fix flagged by Cursor Bugbot on chore: sync develop with main and tidy Python conventions #14.

Closes #13. Supersedes the docs portion of #15.

Test plan

  • Open Colab badge from README — loads the notebook from `main` (not the Drive copy)
  • Run `GenerateCheckWeights.ipynb` end-to-end against a sample PyTorch model — `mymodel_weights.pth` is produced and verifies
  • Run `GenerateCheckWeights.ipynb` end-to-end against a sample TensorFlow model — `mymodel_weights.pkl` is produced and verifies
  • Skim training guide for the wording / variable-name changes

🤖 Generated with Claude Code


Note

Low Risk
Documentation-only changes: updates notebook instructions/links and rewrites a helper notebook to avoid a missing SDK import; no runtime production code paths affected.

Overview
Fixes the broken weights helper notebook. GenerateCheckWeights.ipynb is rewritten to generate and validate weights directly via importlib + torch.save (PyTorch .pth) / pickle (TF .pkl), replacing the stale tracebloc.ModelWeights usage.

Updates Colab entry points and guide UX. README and traceblocTrainingGuide.ipynb now link to the GitHub-backed Colab URL (instead of a Drive copy), and the training guide tweaks wording and variable naming (trainingObjecttraining) for consistency.

Reviewed by Cursor Bugbot for commit 9c640d3. Bugbot is set up for automated code reviews on this repo. Configure here.

…link

Splits the docs portion out of #15. The fix/pr-15-conflicts branch had
correctly resolved these against main; this is the clean cut against
main without the CI workflows or stale merge history.

- notebooks/GenerateCheckWeights.ipynb: rewrite for 0.8.x SDK. The
  previous version imported `from tracebloc import ModelWeights`,
  which no longer exists in 0.8.x — running it on main currently
  raises ImportError. New version uses importlib + torch.save/pickle
  directly, matching the file layout the SDK expects (.pth for
  PyTorch, .pkl for TensorFlow).
- README.md + notebooks/traceblocTrainingGuide.ipynb: repoint the
  Colab badge from the Drive-hosted copy
  (drive/1N00idtpoaq1lk9OJE6g4bMqd8o-Qex2C) to the GitHub-backed URL
  (colab.research.google.com/github/tracebloc/start-training/...).
  The Drive copy required manual sync on every notebook update and
  drifted; the GitHub form loads the canonical notebook from main
  directly.
- notebooks/traceblocTrainingGuide.ipynb: minor UX — training_object
  → training, "Dataset" wording aligned with the use-case panel
  label, Drive-save reminder inlined in Prerequisites, .pth/.pkl
  extension fix flagged by Cursor Bugbot on #14.

Closes #13.

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
@saadqbal saadqbal self-assigned this May 19, 2026
@LukasWodka LukasWodka merged commit 5bf3b6c into main May 19, 2026
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Sync develop with main and finish Python convention cleanup

2 participants