Skip to content

feat: Unified predictive task result schemas#241

Merged
rmitsch merged 4 commits intomainfrom
240
Dec 23, 2025
Merged

feat: Unified predictive task result schemas#241
rmitsch merged 4 commits intomainfrom
240

Conversation

@rmitsch
Copy link
Collaborator

@rmitsch rmitsch commented Dec 23, 2025

Description

Unifies predictive task result schemas across bridges.

Related Issues

Changes Made

  • Unifies result schemas across bridges for predictive tasks
  • Introduce schemas modules in task modules to hold (most) schemas
  • Updates test suite accordingly
  • Updates docs accordingly
  • Include a cached flag in doc-task metadata to indicate whether a doc has been served from cache
  • Fixes a bug in Outlines token counting for transformers models

Checklist

  • Tests have been extended to cover changes in functionality
  • Existing and new tests succeed
  • Documentation updated (if applicable)
  • Related issues linked

Screenshots/Examples (if applicable)

@rmitsch rmitsch self-assigned this Dec 23, 2025
@rmitsch rmitsch marked this pull request as ready for review December 23, 2025 15:08
@codecov
Copy link

codecov bot commented Dec 23, 2025

Codecov Report

❌ Patch coverage is 93.28358% with 36 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
sieves/tasks/predictive/gliner_bridge.py 91.92% 13 Missing ⚠️
sieves/tasks/predictive/ner/schemas.py 85.29% 5 Missing ⚠️
sieves/tasks/predictive/classification/bridges.py 75.00% 4 Missing ⚠️
sieves/tasks/predictive/classification/schemas.py 92.30% 3 Missing ⚠️
...tasks/predictive/information_extraction/schemas.py 92.59% 2 Missing ⚠️
...ves/tasks/predictive/sentiment_analysis/schemas.py 91.66% 2 Missing ⚠️
sieves/tasks/predictive/summarization/bridges.py 71.42% 2 Missing ⚠️
sieves/tasks/predictive/translation/bridges.py 71.42% 2 Missing ⚠️
sieves/tasks/predictive/ner/bridges.py 75.00% 1 Missing ⚠️
sieves/tasks/predictive/pii_masking/schemas.py 95.65% 1 Missing ⚠️
... and 1 more
Additional details and impacted files
@@            Coverage Diff             @@
##             main     #241      +/-   ##
==========================================
+ Coverage   93.21%   93.34%   +0.12%     
==========================================
  Files          62       72      +10     
  Lines        3554     3681     +127     
==========================================
+ Hits         3313     3436     +123     
- Misses        241      245       +4     
Files with missing lines Coverage Δ
sieves/model_wrappers/core.py 90.78% <100.00%> (+0.37%) ⬆️
sieves/model_wrappers/outlines_.py 88.88% <100.00%> (ø)
sieves/pipeline/core.py 93.54% <100.00%> (+0.10%) ⬆️
sieves/tasks/predictive/__init__.py 100.00% <100.00%> (ø)
sieves/tasks/predictive/bridges.py 100.00% <100.00%> (+7.69%) ⬆️
sieves/tasks/predictive/classification/__init__.py 100.00% <100.00%> (ø)
sieves/tasks/predictive/classification/core.py 96.64% <100.00%> (+1.35%) ⬆️
sieves/tasks/predictive/core.py 91.37% <100.00%> (-0.34%) ⬇️
...asks/predictive/information_extraction/__init__.py 100.00% <100.00%> (ø)
...tasks/predictive/information_extraction/bridges.py 96.62% <100.00%> (+0.01%) ⬆️
... and 30 more
🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@rmitsch rmitsch merged commit 83b1e76 into main Dec 23, 2025
3 checks passed
@rmitsch rmitsch deleted the 240 branch December 23, 2025 15:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Unify task output types

1 participant