Skip to content

Cleanup numba dispatches#1648

Merged
ricardoV94 merged 16 commits into
pymc-devs:mainfrom
ricardoV94:cleanup_numba_dispatches
Oct 16, 2025
Merged

Cleanup numba dispatches#1648
ricardoV94 merged 16 commits into
pymc-devs:mainfrom
ricardoV94:cleanup_numba_dispatches

Test numba slice boxing and fix representation of None stop with nega…

c594eb4
Select commit
Loading
Failed to load commit list.
Codecov / codecov/project succeeded Oct 14, 2025 in 1s

81.63% (+0.05%) compared to 602eb04

View this Pull Request on Codecov

81.63% (+0.05%) compared to 602eb04

Details

Codecov Report

❌ Patch coverage is 90.70796% with 21 lines in your changes missing coverage. Please review.
✅ Project coverage is 81.63%. Comparing base (602eb04) to head (c594eb4).

Files with missing lines Patch % Lines
pytensor/link/numba/dispatch/basic.py 77.77% 7 Missing and 3 partials ⚠️
pytensor/link/numba/dispatch/subtensor.py 82.92% 5 Missing and 2 partials ⚠️
pytensor/link/numba/dispatch/elemwise.py 90.47% 3 Missing and 1 partial ⚠️
Additional details and impacted files

Impacted file tree graph

@@            Coverage Diff             @@
##             main    #1648      +/-   ##
==========================================
+ Coverage   81.58%   81.63%   +0.05%     
==========================================
  Files         240      242       +2     
  Lines       53593    53564      -29     
  Branches     9454     9451       -3     
==========================================
+ Hits        43722    43727       +5     
+ Misses       7395     7360      -35     
- Partials     2476     2477       +1     
Files with missing lines Coverage Δ
pytensor/configdefaults.py 73.54% <ø> (-0.09%) ⬇️
pytensor/configparser.py 92.55% <ø> (-0.03%) ⬇️
pytensor/link/numba/dispatch/__init__.py 100.00% <100.00%> (ø)
pytensor/link/numba/dispatch/extra_ops.py 95.51% <100.00%> (+0.02%) ⬆️
pytensor/link/numba/dispatch/scalar.py 87.64% <100.00%> (-2.64%) ⬇️
pytensor/link/numba/dispatch/scan.py 96.02% <ø> (-0.03%) ⬇️
pytensor/link/numba/dispatch/shape.py 100.00% <100.00%> (ø)
pytensor/link/numba/dispatch/sort.py 100.00% <100.00%> (ø)
pytensor/link/numba/dispatch/tensor_basic.py 88.33% <100.00%> (+0.94%) ⬆️
pytensor/link/numba/dispatch/elemwise.py 94.28% <90.47%> (-0.54%) ⬇️
... and 2 more
🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.