rajatsainju2025/code-explainer

Test: Trainer small config

Aperta

#11 aperta il 10 ago 2025

 (1 commento) (0 reazioni) (0 assegnatari)Python (0 fork)auto 404
good first issue

Metriche repository

Star
 (1 stella)
Metriche merge PR
 (Nessuna PR mergiata in 30 g)

Descrizione

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

Guida contributor