rajatsainju2025/code-explainer

Test: Trainer small config

Open

#11 opened on Aug 10, 2025

View on GitHubContributor guide
 (1 comment) (0 reactions) (0 assignees)Python (0 forks)auto 404
good first issue

Repository metrics

Stars
 (1 star)
PR merge metrics
 (PR metrics pending)

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

Contributor guide

Research direction
Examine existing tests in tests/test trainer.py for patterns, then create a new test file that loads sshleifer/tiny gpt2, builds a minimal config, and calls the trainer methods. Ensure the test runs quickly and does not perform full training.
Tech stack
python
Domain
backend
Issue type
Test
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
1
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
Under 1 hour
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
Active
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
Clear
Prerequisites
GitPython
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
90