Spaces:
Running
Running
| import pandas as pd | |
| from src.evaluate.eval import evaluate_outputs | |
| def test_evaluate_outputs_adds_metrics(): | |
| df = pd.DataFrame([{ | |
| "Description": "Crash blocked one lane on I-95 northbound.", | |
| "reference_summary": "One lane was blocked after a crash on I-95 northbound.", | |
| "generated_summary": "A crash blocked one lane on I-95 northbound.", | |
| "model_name": "lead1", | |
| }]) | |
| out = evaluate_outputs(df) | |
| assert "compression_ratio" in out.columns | |
| assert "rouge1_f1" in out.columns | |