rajatsainju2025/code-explainer

Test: Trainer small config

开放

#11 创建于 2025年8月10日

 (1 条评论) (0 个反应) (0 位负责人)Python (0 个派生)auto 404
good first issue

仓库指标

星标
 (1 个星标)
PR 合并指标
 (PR 指标待抓取)

描述

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

贡献者指南