CodeT5p-220m PyTest Edge-Case Generator

This model is a fine-tuned version of Salesforce/codet5p-220m, starting from my normal-test fine-tuned checkpoint, for generating PyTest unit tests focused on edge cases and boundary conditions.

Overview

It prioritizes:

  • empty inputs
  • None handling
  • extreme values
  • invalid types
  • exceptions (pytest.raises)
  • boundary conditions

Training results

Eval loss by epoch:

Epoch Eval Loss
1 1.8182
2 1.7890
3 1.7791
4 1.7804
  • Best eval_loss: 1.7791306972503662

Citation

@misc{akinwale2025codet5pytestedge,
  author  = {Akinwale, Oluwaferanmi Philip},
  title   = {CodeT5p-220m PyTest Edge-Case Generator},
  year    = {2025},
  note    = {Hugging Face model repository},
  url     = {https://huggingface.co/Oluwaferanmiii/codet5p-220m-pytest-edge-generator}
}
Downloads last month
4
Safetensors
Model size
0.2B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for Oluwaferanmiii/codet5p-220m-pytest-edge-generator

Finetuned
(93)
this model

Evaluation results

  • Best eval loss on Custom edge-case function–test pairs
    self-reported
    1.779