[BREAKING] Drop TensorFlow backend by felixT2K · Pull Request #1967 · mindee/doctr (original) (raw)
Conversation
felixT2K commented
•
edited by felixdittrich92
Loading
This PR:
- Remove all TensorFlow content
- Add PyTorch as fixed dependency
- Drop documentation versions which rely on TensorFlow backend
- Remove conda which does not support PyTorch builds anymore - since 2.15
- Remove unused code for channels_last / first switch
- Fix typings & unify predictor & model inputs - predictors (numpy array HWC) - direct model calls (torch tensor CHW)
Codecov Report
All modified and coverable lines are covered by tests ✅
Project coverage is 96.27%. Comparing base (d01acdd) to head (d297a9c).
Report is 2 commits behind head on main.
Additional details and impacted files
@@ Coverage Diff @@ ## main #1967 +/- ##
Coverage 96.93% 96.27% -0.66%
Files 172 141 -31
Lines 8633 6037 -2596- Hits 8368 5812 -2556
- Misses 265 225 -40
| Flag | Coverage Δ | |
|---|---|---|
| unittests | 96.27% <100.00%> (-0.66%) | ⬇️ |
Flags with carried forward coverage won't be shown. Click here to find out more.
☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.
🚀 New features to boost your workflow:
- ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.
NOTE: API test will fail until merged
felixT2K marked this pull request as ready for review
@sebastianMindee That should be everything for a first round to get rid of TensorFlow 😓 😅
API test should be green after it's merged
This was referenced
Jul 9, 2025
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.Learn more about bidirectional Unicode characters
[ Show hidden characters]({{ revealButtonHref }})