rajatsainju2025/code-explainer

Test: Trainer small config

Aberta

#11 aberto em 10 de ago. de 2025

 (1 comentário) (0 reação) (0 responsável)Python (0 fork)auto 404
good first issue

Métricas do repositório

Stars
 (1 estrela)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

Description

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

Guia do colaborador