rajatsainju2025/code-explainer

Test: Trainer small config

Offen

#11 geöffnet am 10.08.2025

 (1 Kommentar) (0 Reaktionen) (0 zugewiesene Personen)Python (0 Forks)auto 404
good first issue

Repository-Metriken

Stars
 (1 Stern)
PR-Merge-Metriken
 (Keine gemergten PRs in 30 T)

Beschreibung

Summary

Add a unit test to run the trainer with a very small config using a tiny model, ensuring that model loading, dataset preprocessing, and trainer setup succeed.

Acceptance Criteria

  • A new test file tests/test_trainer_small_config.py that:
    • Uses sshleifer/tiny-gpt2
    • Builds a minimal JSON config
    • Calls CodeExplainerTrainer.load_model(), load_dataset(None), preprocess_dataset(), and setup_trainer()
  • Test should run quickly (< 60s) and not perform full training.

References

  • Existing tests in tests/test_trainer.py
  • Tiny models used in CI

Contributor Guide